diff --git a/unsloth/import_fixes.py b/unsloth/import_fixes.py
index 3ba79c2162..75dc12f5a1 100644
--- a/unsloth/import_fixes.py
+++ b/unsloth/import_fixes.py
@@ -13,6 +13,7 @@
 # limitations under the License.
 
 import os
+import sys
 import importlib.util
 from pathlib import Path
 from importlib.metadata import version as importlib_version