diff --git a/unsloth/import_fixes.py b/unsloth/import_fixes.py
index 23bb4789ca..3b10ec26fb 100644
--- a/unsloth/import_fixes.py
+++ b/unsloth/import_fixes.py
@@ -19,6 +19,7 @@ from importlib.metadata import version as importlib_version
 from packaging.version import Version as TrueVersion
 import re
 import logging
+import textwrap
 
 # We cannot do from unsloth_zoo.log import logger since FBGEMM might cause seg faults.
 UNSLOTH_ENABLE_LOGGING = os.environ.get("UNSLOTH_ENABLE_LOGGING",  "0") in ("1", "True", "true",)