diff --git a/plm_interpretability/sae_model.py b/plm_interpretability/sae_model.py
index c79af51..9b8f1bd 100644
--- a/plm_interpretability/sae_model.py
+++ b/plm_interpretability/sae_model.py
@@ -7,7 +7,8 @@
 import torch.nn as nn
 from torch.nn import functional as F
 from transformers import PreTrainedModel, PreTrainedTokenizer
-from utils import get_layer_activations
+
+from plm_interpretability.utils import get_layer_activations
 
 
 class SparseAutoencoder(nn.Module):