diff --git a/monkey/tests/unit_tests/infection_monkey/island_api_client/test_http_island_api_client.py b/monkey/tests/unit_tests/infection_monkey/island_api_client/test_http_island_api_client.py
index e65ba4eba..eb8ae5df7 100644
--- a/monkey/tests/unit_tests/infection_monkey/island_api_client/test_http_island_api_client.py
+++ b/monkey/tests/unit_tests/infection_monkey/island_api_client/test_http_island_api_client.py
@@ -21,7 +21,7 @@ from infection_monkey.island_api_client import (
     IslandAPITimeoutError,
 )
 
-SERVER = SocketAddress(ip="1.1.1.1", port="9999")
+SERVER = SocketAddress(ip="1.1.1.1", port=9999)
 PBA_FILE = "dummy.pba"
 WINDOWS = "windows"
 AGENT_ID = UUID("80988359-a1cd-42a2-9b47-5b94b37cd673")
diff --git a/monkey/tests/unit_tests/monkey_island/cc/island_event_handlers/test_handle_agent_registration.py b/monkey/tests/unit_tests/monkey_island/cc/island_event_handlers/test_handle_agent_registration.py
index dff59f5d3..316e4d8c0 100644
--- a/monkey/tests/unit_tests/monkey_island/cc/island_event_handlers/test_handle_agent_registration.py
+++ b/monkey/tests/unit_tests/monkey_island/cc/island_event_handlers/test_handle_agent_registration.py
@@ -114,7 +114,7 @@ def test_existing_machine_updated__find_by_ip(handler, machine_repository):
         machine_hardware_id=5,
         start_time=0,
         parent_id=None,
-        cc_server=SocketAddress(ip="192.168.1.1", port="5000"),
+        cc_server=SocketAddress(ip="192.168.1.1", port=5000),
         network_interfaces=[
             IPv4Interface("192.168.1.2/24"),
             IPv4Interface("192.168.1.4/24"),
@@ -218,7 +218,7 @@ def test_machine_interfaces_updated(handler, machine_repository):
         machine_hardware_id=MACHINE.hardware_id,
         start_time=0,
         parent_id=None,
-        cc_server=SocketAddress(ip="192.168.1.1", port="5000"),
+        cc_server=SocketAddress(ip="192.168.1.1", port=5000),
         network_interfaces=[
             IPv4Interface("192.168.1.2/24"),
             IPv4Interface("192.168.1.3/16"),
diff --git a/monkey/tests/unit_tests/monkey_island/cc/models/test_agent_registration_message.py b/monkey/tests/unit_tests/monkey_island/cc/models/test_agent_registration_message.py
index de51fefd2..dff49c252 100644
--- a/monkey/tests/unit_tests/monkey_island/cc/models/test_agent_registration_message.py
+++ b/monkey/tests/unit_tests/monkey_island/cc/models/test_agent_registration_message.py
@@ -10,7 +10,7 @@ from common.types import SocketAddress
 
 AGENT_ID = UUID("012e7238-7b81-4108-8c7f-0787bc3f3c10")
 PARENT_ID = UUID("0fc9afcb-1902-436b-bd5c-1ad194252484")
-SOCKET_ADDRESS = SocketAddress(ip="192.168.1.1", port="5000")
+SOCKET_ADDRESS = SocketAddress(ip="192.168.1.1", port=5000)
 
 AGENT_REGISTRATION_MESSAGE_OBJECT_DICT = {
     "id": AGENT_ID,
diff --git a/monkey/tests/unit_tests/monkey_island/cc/repository/test_mongo_agent_repository.py b/monkey/tests/unit_tests/monkey_island/cc/repository/test_mongo_agent_repository.py
index aa804cd13..62ebaed1c 100644
--- a/monkey/tests/unit_tests/monkey_island/cc/repository/test_mongo_agent_repository.py
+++ b/monkey/tests/unit_tests/monkey_island/cc/repository/test_mongo_agent_repository.py
@@ -55,7 +55,7 @@ AGENTS = (
     *RUNNING_AGENTS,
     *STOPPED_AGENTS,
 )
-CC_SERVER = SocketAddress(ip="127.0.0.1", port="1984")
+CC_SERVER = SocketAddress(ip="127.0.0.1", port=1984)
 
 
 @pytest.fixture