diff --git a/localization_log_gamma_model.py b/localization_log_gamma_model.py
index 43b7637..b4ea53f 100644
--- a/localization_log_gamma_model.py
+++ b/localization_log_gamma_model.py
@@ -32,7 +32,7 @@ def fittingSingle(filepath, args):
         print("err: failed to load file {}".format(filepath))
         return
     location_gt_tmp = None
-    if 'img_' in os.path.basename(filepath):
+    if 'img_' == os.path.basename(filepath)[:4]:
         location_gt_tmp = os.path.basename(filepath).split('_')
         location_gt = [float(location_gt_tmp[1]) + 3.2, float(location_gt_tmp[2]) + 3.2]
     else: