diff --git a/README.md b/README.md
index 5b6595a..55383dd 100644
--- a/README.md
+++ b/README.md
@@ -172,7 +172,7 @@ This is useful for integrating MFLUX into shell scripts or dynamically generatin
 Alternatively, you can use MFLUX directly in Python:
 
 ```python
-from mflux import Flux1
+from mflux.models.flux.variants.txt2img.flux import Flux1
 
 # Load the model
 flux = Flux1.from_name(
diff --git a/src/mflux/__init__.py b/src/mflux/__init__.py
index 9795341..c72ee76 100644
--- a/src/mflux/__init__.py
+++ b/src/mflux/__init__.py
@@ -4,8 +4,3 @@ import os
 # This must be set before any tokenizers are imported/used
 if "TOKENIZERS_PARALLELISM" not in os.environ:
     os.environ["TOKENIZERS_PARALLELISM"] = "false"
-
-# Export main classes for convenient import
-from mflux.models.flux.variants.txt2img.flux import Flux1
-
-__all__ = ["Flux1"]