diff --git a/deepmosaic.py b/deepmosaic.py
index 7f2b4db4a9d1ab486fd60281696062d6793b154c..cfc871757cb71d0e071451e2971b4251ddaddef2 100644
--- a/deepmosaic.py
+++ b/deepmosaic.py
@@ -1,9 +1,14 @@
 import os
 import sys
 import traceback
-from cores import Options,core
-from util import util
-from models import loadmodel
+try:
+    from cores import Options,core
+    from util import util
+    from models import loadmodel
+except Exception as e:
+    print(e)
+    input('Please press any key to exit.\n')
+    sys.exit(0)
 
 opt = Options().getparse(test_flag = True)
 if not os.path.isdir(opt.temp_dir):