diff --git a/pyvene/__init__.py b/pyvene/__init__.py
index 2527fa4a..8e2934fd 100644
--- a/pyvene/__init__.py
+++ b/pyvene/__init__.py
@@ -52,4 +52,4 @@
 from .models.llama.modelings_intervenable_llama import create_llama
 from .models.mlp.modelings_intervenable_mlp import create_mlp_classifier
 from .models.backpack_gpt2.modelings_intervenable_backpack_gpt2 import create_backpack_gpt2
-
+from .models.olmo.modelings_intervenable_olmo import create_olmo