diff --git a/lib/tools.py b/lib/tools.py
index 071c295..00fdc6d 100755
--- a/lib/tools.py
+++ b/lib/tools.py
@@ -141,9 +141,9 @@ def __init__(self, checkpoint, use_cuda, half_precision, im_fe_ratio = 16, postp
 
         self.model = ImMatchNet(use_cuda=use_cuda, multi_gpu=multi_gpu, half_precision=half_precision, checkpoint=checkpoint)
         self.model.eval()
-        self.feature_extractor = ExtractFeatureMap(im_fe_ratio=self.im_fe_ratio, device=postprocess_device)
         self.postprocess_device = postprocess_device
         self.im_fe_ratio = im_fe_ratio
+        self.feature_extractor = ExtractFeatureMap(im_fe_ratio=self.im_fe_ratio, device=postprocess_device)
 
     def __call__(self, batch, num_pts=2000, iter_step=1000, central_align = True, *args, **kwargs):
         '''