diff --git a/unsloth/save.py b/unsloth/save.py
index ea2d30972e..11e1e7a3fb 100644
--- a/unsloth/save.py
+++ b/unsloth/save.py
@@ -12,6 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from packaging.version import Version
 from bitsandbytes.nn import Linear4bit as Bnb_Linear4bit
 from peft.tuners.lora import Linear4bit as Peft_Linear4bit
 from peft.tuners.lora import Linear as Peft_Linear