diff --git a/tests/unit/script/types/test_function.py b/tests/unit/script/types/test_function.py
index 014e4a85..e13828f4 100644
--- a/tests/unit/script/types/test_function.py
+++ b/tests/unit/script/types/test_function.py
@@ -49,7 +49,11 @@ class TestFunction:
     @pytest.mark.parametrize(
         "function_str, expected_types, received_types",
         [
-            ("{%array_at({'a': 'dict?'}, 1)}", "array: Array, idx: Integer", "Map, Integer"),
+            (
+                "{%array_at({'a': 'dict?'}, 1)}",
+                "array: Array, idx: Integer, default: Optional[AnyArgument]",
+                "Map, Integer",
+            ),
             ("{%array_extend('not', 'array')}", "arrays: Array, ...", "String, String"),
             (
                 "{%replace('hi mom', 'mom', 'dad', 1, 0)}",