# 2041: Minor test changes

src/primaite/simulator/system/services/ntp/ntp_client.py

@@ -100,9 +100,9 @@ class NTPClient(Service):
             self.time = payload.ntp_reply.ntp_datetime
             return True
 
-    def request_time(self) -> None:
+    def request_time(self, ip_address: IPv4Address = ip_addr) -> None:
         """Send request to ntp_server."""
-        ntp_request = NTPRequest(ntp_client=self.ip_addr)
+        ntp_request = NTPRequest(ntp_client=ip_address)
         ntp_server_packet = NTPPacket(ntp_request=ntp_request)
         self.send(payload=ntp_server_packet)
 

src/primaite/simulator/system/services/ntp/ntp_server.py

@@ -1,5 +1,5 @@
 from datetime import datetime
-from typing import Any, Dict, Optional
+from typing import Dict, Optional
 
 from primaite import getLogger
 from primaite.simulator.network.protocols.ntp import NTPPacket
@@ -46,7 +46,7 @@ class NTPServer(Service):
 
     def receive(
         self,
-        payload: Any,
+        payload: NTPPacket,
         session_id: Optional[str] = None,
         **kwargs,
     ) -> bool: