diff --git a/klient/client.py b/klient/client.py
index a64ddba..8cb15da 100644
--- a/klient/client.py
+++ b/klient/client.py
@@ -76,7 +76,7 @@ def get_command(args) -> List[str]:
 
 
 def run_subprocess(
-    command: List[str], state_json: str, timeout: Optional[float]
+    command: List[str], state_json: str, timeout: Optional[int]
 ) -> Optional[str]:
     """Run user program and return its output."""
 
diff --git a/klient/play.py b/klient/play.py
index b1667b9..4c68f0c 100755
--- a/klient/play.py
+++ b/klient/play.py
@@ -34,6 +34,7 @@ class State:
         self.world = parse_world(state["state"]["world"])
         self.my_team_id: int = state["team_id"]
         self.round_number: int = state["round"]
+        self.time_to_response: Optional[int] = state["time_to_response"]
 
 
 state: State