diff --git a/pulmonary_embolism.py b/pulmonary_embolism.py
index fe463c4..5dc4314 100644
--- a/pulmonary_embolism.py
+++ b/pulmonary_embolism.py
@@ -14,7 +14,7 @@
 import shutil
 import matplotlib as mpl
 
-mpl.use('TkAgg')
+#mpl.use('TkAgg')
 import matplotlib.pyplot as plt
 
 # Root directory of the project
@@ -358,7 +358,7 @@ def get_ax(rows=1, cols=1, size=8):
         MODEL_PATH = args.model
         print("Loading weights from ", MODEL_PATH)
 
-        model.load_weights(COCO_MODEL_PATH, by_name=True,
+        model.load_weights(MODEL_PATH, by_name=True,
                            exclude=["mrcnn_class_logits", "mrcnn_bbox_fc",
                                     "mrcnn_bbox", "mrcnn_mask"])
         print("finish")