diff --git a/bardapi/models/tools/map.py b/bardapi/models/tools/map.py
index 6bdf502ce..bdaeb0591 100644
--- a/bardapi/models/tools/map.py
+++ b/bardapi/models/tools/map.py
@@ -1,4 +1,4 @@
-from typing import Optional
+from typing import List, Optional, Union, Dict, Tuple
 
 from bardapi.models.user_content import UserContent