diff --git a/test/integration/test_groundlight.py b/test/integration/test_groundlight.py
index ab82f0e7..3e183263 100644
--- a/test/integration/test_groundlight.py
+++ b/test/integration/test_groundlight.py
@@ -20,6 +20,7 @@
     CountingResult,
     Detector,
     ImageQuery,
+    MultiClassificationResult,
     PaginatedDetectorList,
     PaginatedImageQueryList,
 )
@@ -30,7 +31,11 @@
 
 def is_valid_display_result(result: Any) -> bool:
     """Is the image query result valid to display to the user?."""
-    if not isinstance(result, BinaryClassificationResult) and not isinstance(result, CountingResult):
+    if (
+        not isinstance(result, BinaryClassificationResult)
+        and not isinstance(result, CountingResult)
+        and not isinstance(result, MultiClassificationResult)
+    ):
         return False
     if not is_valid_display_label(result.label):
         return False
diff --git a/test/unit/test_internalapi.py b/test/unit/test_internalapi.py
index fc207509..e6f6fbca 100644
--- a/test/unit/test_internalapi.py
+++ b/test/unit/test_internalapi.py
@@ -4,11 +4,11 @@
 
 def test_iq_is_confident(gl_experimental: ExperimentalApi):
     det = gl_experimental.get_or_create_detector("Test", "test_query")
-    iq = gl_experimental.ask_async(det, image="test/assets/dog.jpeg", wait=10)
+    iq = gl_experimental.ask_async(det, image="test/assets/dog.jpeg")
     assert not iq_is_confident(iq, 0.9)
 
 
 def test_iq_is_answered(gl_experimental: ExperimentalApi):
     det = gl_experimental.get_or_create_detector("Test", "test_query")
-    iq = gl_experimental.ask_async(det, image="test/assets/dog.jpeg", wait=10)
+    iq = gl_experimental.ask_async(det, image="test/assets/dog.jpeg")
     assert not iq_is_answered(iq)