diff --git a/sage_client.py b/sage_client.py
index 7aa034a..64208ae 100644
--- a/sage_client.py
+++ b/sage_client.py
@@ -1,5 +1,5 @@
 import requests
-from typing import Dict, List, Optional
+from typing import Dict, List, Optional, Union
 from config import settings
 import logging