diff --git a/pyinfra/visitor.py b/pyinfra/visitor.py
index 55a8a87..155d5bd 100644
--- a/pyinfra/visitor.py
+++ b/pyinfra/visitor.py
@@ -58,7 +58,7 @@ class StorageStrategy(ResponseStrategy):
     def __init__(self, storage):
         self.storage = storage
 
-    def handle_response(self, body):
+    def handle_response(self, body: dict):
         response_object_descriptor = self.get_response_object_descriptor(body)
         self.storage.put_object(**response_object_descriptor, data=gzip.compress(json.dumps(body).encode()))
         body.pop("data")