diff --git a/gguf-py/gguf/__init__.py b/gguf-py/gguf/__init__.py
index 40b77128e..f9b70a85b 100644
--- a/gguf-py/gguf/__init__.py
+++ b/gguf-py/gguf/__init__.py
@@ -1 +1 @@
-from .gguf import GGUFWriter, get_tensor_name_map, MODEL_ARCH
+from .gguf import *