diff --git a/eval_image_retrieval.py b/eval_image_retrieval.py
index 999f8c900..b6bdc20b2 100644
--- a/eval_image_retrieval.py
+++ b/eval_image_retrieval.py
@@ -40,8 +40,7 @@ def __init__(self, dir_main, dataset, split, transform=None, imsize=None):
         with open(gnd_fname, 'rb') as f:
             cfg = pickle.load(f)
         cfg['gnd_fname'] = gnd_fname
-        cfg['ext'] = '.jpg'
-        cfg['qext'] = '.jpg'
+        cfg['qext'] = cfg['ext'] = '.jpg'
         cfg['dir_data'] = os.path.join(dir_main, dataset)
         cfg['dir_images'] = os.path.join(cfg['dir_data'], 'jpg')
         cfg['n'] = len(cfg['imlist'])