diff --git a/hopfield/module.py b/hopfield/module.py
index d647d9b..a3c67a9 100644
--- a/hopfield/module.py
+++ b/hopfield/module.py
@@ -13,7 +13,7 @@ from hflayers import Hopfield
 from hflayers.transformer import HopfieldDecoderLayer, HopfieldEncoderLayer
 
 
-class HopfieldNetwork(nn.Module):
+class HopfieldModel(nn.Module):
     @validated()
     def __init__(
         self,