diff --git a/deel/torchlip/__init__.py b/deel/torchlip/__init__.py
index be2c5b0..72c0e52 100644
--- a/deel/torchlip/__init__.py
+++ b/deel/torchlip/__init__.py
@@ -55,4 +55,5 @@
     "Sequential",
     "SpectralConv2d",
     "SpectralLinear",
+    "vanilla_model",
 ]
diff --git a/deel/torchlip/modules/__init__.py b/deel/torchlip/modules/__init__.py
index b2653d3..7f730d4 100644
--- a/deel/torchlip/modules/__init__.py
+++ b/deel/torchlip/modules/__init__.py
@@ -64,6 +64,7 @@
 from .loss import NegKRLoss
 from .module import LipschitzModule
 from .module import Sequential
+from .module import vanilla_model
 from .pooling import ScaledAdaptiveAvgPool2d
 from .pooling import ScaledAvgPool2d
 from .pooling import ScaledL2NormPool2d