diff --git a/unsloth/models/gemma.py b/unsloth/models/gemma.py
index 22734a0933..51432f337b 100644
--- a/unsloth/models/gemma.py
+++ b/unsloth/models/gemma.py
@@ -57,7 +57,7 @@ def fast_geglu_inference(self, X):
 pass
 
 
-class FastGemmaRotaryEmbedding(nn.Module):
+class FastGemmaRotaryEmbedding(torch.nn.Module):
     def __init__(self, dim, max_position_embeddings=2048, base=10000, device=None):
         super().__init__()