diff --git a/unittests/test_log.py b/unittests/test_log.py
index 67c04368..aad534f7 100644
--- a/unittests/test_log.py
+++ b/unittests/test_log.py
@@ -72,7 +72,7 @@ def retreive_log_messages(
         return result
 
     @overrides(LogStore.get_location)
-    def get_location(self):
+    def get_location(self) -> str:
         return "MOCK"