diff --git a/tests/baseline.json b/tests/baseline.json
index c30f17ac5..bde8b2093 100644
--- a/tests/baseline.json
+++ b/tests/baseline.json
@@ -4,8 +4,8 @@
      "extension_count": 13,
      "function_count": 165,
      "num_aggregate_functions": 29,
-     "num_scalar_functions": 169,
-     "num_window_functions": 0,
+     "num_scalar_functions": 158,
+     "num_window_functions": 11,
      "num_function_overloads": 517
    },
    "coverage": {
diff --git a/tests/coverage/extensions.py b/tests/coverage/extensions.py
index e66681d27..6c8a44f15 100644
--- a/tests/coverage/extensions.py
+++ b/tests/coverage/extensions.py
@@ -188,7 +188,7 @@ def read_substrait_extensions(dir_path: str):
                 if "window_functions" in data:
                     Extension.add_functions_to_map(
                         data["window_functions"],
-                        scalar_functions,
+                        window_functions,
                         suffix,
                         extension,
                         uri,