diff --git a/py/server/tests/test_udf_return.py b/py/server/tests/test_udf_return.py
index ddbfb592fc8..199f34d0dcb 100644
--- a/py/server/tests/test_udf_return.py
+++ b/py/server/tests/test_udf_return.py
@@ -4,7 +4,7 @@
 import datetime
 import typing
 import unittest
-from typing import List, Union, Optional
+from typing import List, Union, Tuple, Sequence, Optional
 
 import numpy as np
 import numpy.typing as npt