diff --git a/shopping_client/shopping_client.py b/shopping_client/shopping_client.py
index 82e6b20754bda6e2f806583534cbd80136af6610..c39b78c8d454db72ab7ef0aaa015d418015c1b6c 100644
--- a/shopping_client/shopping_client.py
+++ b/shopping_client/shopping_client.py
@@ -38,7 +38,7 @@ class shopping_client:
 
     def get_basket(
         self, convert_to_pandas: bool = False, reload: bool = False
-    ) -> Optional[list[dict]]:
+    ) -> Union[list[dict], pd.DataFrame, None]:
         """Retrieve the shopping basket for a user.
         Prompts for access token if one was not supplied to constructor.