diff --git a/intel_extension_for_pytorch/__init__.py b/intel_extension_for_pytorch/__init__.py
index 826562117..3ba61015e 100644
--- a/intel_extension_for_pytorch/__init__.py
+++ b/intel_extension_for_pytorch/__init__.py
@@ -1,6 +1,7 @@
 # This Python file uses the following encoding: utf-8
 import re
 import os
+import sys
 import torch
 
 
@@ -33,7 +34,7 @@
         + f"{ipex_version}.*, but PyTorch {torch.__version__} is found. "
         + "Please switch to the matching version and run again."
     )
-    os.exit(127)
+    sys.exit(127)
 
 
 import os