diff --git a/deepspec/utils/constant/__init__.py b/deepspec/utils/constant/__init__.py
index c887caeb..43f42dc2 100644
--- a/deepspec/utils/constant/__init__.py
+++ b/deepspec/utils/constant/__init__.py
@@ -1,5 +1,5 @@
 try:
     import hfai
     from .internal import *
-except:
+except (ImportError, ModuleNotFoundError):
     from .public import *
\ No newline at end of file