diff --git a/python/restbus/restbus.py b/python/restbus/restbus.py
index f72ca19..f3024fc 100644
--- a/python/restbus/restbus.py
+++ b/python/restbus/restbus.py
@@ -4,7 +4,7 @@
 import math
 import re
 import time
-from typing import Any, Generator, Iterable, Optional, Tuple
+from typing import Any, Generator, Iterable, Optional, Tuple, List, Dict
 from typing_extensions import TypeAlias
 
 import google.protobuf.internal.containers  # type: ignore