diff --git a/src/aiia/model/Model.py b/src/aiia/model/Model.py
index a9584fa..771caf8 100644
--- a/src/aiia/model/Model.py
+++ b/src/aiia/model/Model.py
@@ -80,8 +80,9 @@ class AIIABaseShared(AIIA):
         
         # Initialize max pooling layer
         self.max_pool = nn.MaxPool2d(
-            kernel_size=self.config.kernel_size,
-            padding=1  # Using same padding as in Conv2d layers
+            kernel_size=1,
+            stride=1,
+            padding=1
         )
 
     def forward(self, x):
@@ -117,7 +118,7 @@ class AIIABase(AIIA):
                 nn.Conv2d(in_channels, self.config.hidden_size,
                           kernel_size=self.config.kernel_size, padding=1),
                 getattr(nn, self.config.activation_function)(),
-                nn.MaxPool2d(kernel_size=2)
+                nn.MaxPool2d(kernel_size=1, stride=1)
              ])
             in_channels = self.config.hidden_size