diff --git a/vat_pytorch/__init__.py b/vat_pytorch/__init__.py
index 83c8eca..29828b9 100644
--- a/vat_pytorch/__init__.py
+++ b/vat_pytorch/__init__.py
@@ -2,4 +2,4 @@
 from .alice import ALICELoss
 from .alicepp import ALICEPPLoss, ALICEPPModule
 from .utils import kl_loss, sym_kl_loss, js_loss, inf_norm, default
-from .modules import *
\ No newline at end of file
+from .models import *
\ No newline at end of file