diff --git a/pyproject.toml b/pyproject.toml
index 7f24aabbf2..808a956c89 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -32,6 +32,10 @@ include-package-data = false
 exclude = ["images*", "tests*"]
 
 [project.optional-dependencies]
+dev = [
+    "pytest",
+]
+
 triton = [
     "triton-windows ; platform_system == 'Windows'",
 ]
diff --git a/tests/test_model_registry.py b/tests/test_model_registry.py
index c3eb4b0c8d..183edc92d5 100644
--- a/tests/test_model_registry.py
+++ b/tests/test_model_registry.py
@@ -11,8 +11,7 @@ from unsloth.model_registry import (
     get_qwen_models,
     get_qwen_vl_models,
 )
-
-from .utils.hf_hub import get_model_info
+from unsloth.utils.hf_hub import get_model_info
 
 MODEL_NAMES = [
     "llama",
diff --git a/unsloth/utils/__init__.py b/unsloth/utils/__init__.py
new file mode 100644
index 0000000000..e69de29bb2
diff --git a/tests/utils/hf_hub.py b/unsloth/utils/hf_hub.py
similarity index 100%
rename from tests/utils/hf_hub.py
rename to unsloth/utils/hf_hub.py