diff --git a/py/guided_model.py b/py/guided_model.py
index 062f50e..2e76d15 100644
--- a/py/guided_model.py
+++ b/py/guided_model.py
@@ -14,7 +14,7 @@ class JankDiffuseHighModelWrapper:
 
     def __getattr__(self, key):
         try:
-            return getattr(self.__guided_model, key)
+            return getattr(self.__guided_model.dhso.model, key)
         except AttributeError:
             raise AttributeError(key) from None