diff --git a/HNS/MB/ShipField.py b/HNS/MB/ShipField.py
index fe38246..6ecd42f 100644
--- a/HNS/MB/ShipField.py
+++ b/HNS/MB/ShipField.py
@@ -156,9 +156,9 @@ class ShipField:
                     break
 
         self.set_ship_direction(ship_direction)
-        self.set_ship_size(ship_size)
+        self.set_ship_size(self.ship_size)
 
-        self.ships.append(ship_size)
+        self.ships.append(self.ship_size)
 
     def shoot(self, row, col):
         if row < 0 or row > self.field_size - 1: