diff --git a/delphi/sparse_coders/sparse_model.py b/delphi/sparse_coders/sparse_model.py
index 0b7654dc..f8df3700 100644
--- a/delphi/sparse_coders/sparse_model.py
+++ b/delphi/sparse_coders/sparse_model.py
@@ -6,7 +6,7 @@
 
 from delphi.config import RunConfig
 
-from .custom.gemmascope import load_gemma_autoencoders
+from .custom.gemmascope import load_gemma_autoencoders, load_gemma_hooks
 from .load_sparsify import (
     PotentiallyWrappedSparseCoder,
     load_sparsify_hooks,
@@ -58,7 +58,7 @@ def load_hooks_sparse_coders(
             sae_sizes.append(sae_size)
             l0s.append(l0)
 
-        hookpoint_to_sparse_encode = load_gemma_autoencoders(
+        hookpoint_to_sparse_encode = load_gemma_hooks(
             model_path=model_path,
             ae_layers=layers,
             average_l0s=l0s,