diff --git a/tests/resources/stubs/market_service_stub.py b/tests/resources/stubs/market_service_stub.py
index 733afa9a..2e4c3337 100644
--- a/tests/resources/stubs/market_service_stub.py
+++ b/tests/resources/stubs/market_service_stub.py
@@ -1,3 +1,4 @@
+from typing import Dict
 from datetime import datetime
 from random import randint
 
@@ -33,11 +34,11 @@ def orders(self, value):
         self._orders = value
 
     @property
-    def balances(self) -> dict[str, float]:
+    def balances(self) -> Dict[str, float]:
         return self._balances
 
     @balances.setter
-    def balances(self, value: dict[str, float]):
+    def balances(self, value: Dict[str, float]):
         self._balances = value
 
     @property
@@ -150,7 +151,7 @@ def create_limit_sell_order(
             trading_symbol=trading_symbol,
         )
 
-    def get_balance(self, market) -> dict[str, float]:
+    def get_balance(self, market) -> Dict[str, float]:
         return self._balances
 
     def get_order(self, order, market):