diff --git a/monkey/tests/unit_tests/infection_monkey/master/test_propagator.py b/monkey/tests/unit_tests/infection_monkey/master/test_propagator.py
index b662462ac..174296738 100644
--- a/monkey/tests/unit_tests/infection_monkey/master/test_propagator.py
+++ b/monkey/tests/unit_tests/infection_monkey/master/test_propagator.py
@@ -4,6 +4,7 @@ from unittest.mock import MagicMock
 
 import pytest
 
+from common import OperatingSystem
 from common.agent_configuration.agent_sub_configurations import (
     NetworkScanConfiguration,
     PropagationConfiguration,
@@ -33,7 +34,7 @@ def mock_victim_host_factory():
 empty_fingerprint_data = FingerprintData(None, None, {})
 
 dot_1_scan_results = IPScanResults(
-    PingScanData(True, "windows"),
+    PingScanData(True, OperatingSystem.WINDOWS),
     {
         22: PortScanData(22, PortStatus.CLOSED, None, None),
         445: PortScanData(445, PortStatus.OPEN, "SMB BANNER", "tcp-445"),
@@ -47,7 +48,7 @@ dot_1_scan_results = IPScanResults(
 )
 
 dot_3_scan_results = IPScanResults(
-    PingScanData(True, "linux"),
+    PingScanData(True, OperatingSystem.LINUX),
     {
         22: PortScanData(22, PortStatus.OPEN, "SSH BANNER", "tcp-22"),
         443: PortScanData(443, PortStatus.OPEN, "HTTPS BANNER", "tcp-443"),