diff --git a/src/careamics/models/unet.py b/src/careamics/models/unet.py
index 469e09df..7dccae9b 100644
--- a/src/careamics/models/unet.py
+++ b/src/careamics/models/unet.py
@@ -102,7 +102,7 @@ def __init__(
                 )
             )
             encoder_blocks.append(self.pooling)
-        self.encoder_blocks = nn.Modulelist(encoder_blocks)
+        self.encoder_blocks = nn.ModuleList(encoder_blocks)
 
     def forward(self, x: torch.Tensor) -> list[torch.Tensor]:
         """
@@ -222,7 +222,7 @@ def __init__(
                 )
             )
 
-        self.decoder_blocks = nn.Modulelist(decoder_blocks)
+        self.decoder_blocks = nn.ModuleList(decoder_blocks)
 
     def forward(self, *features: torch.Tensor) -> torch.Tensor:
         """