diff --git a/tests/core/engine_adapter/integration/test_integration.py b/tests/core/engine_adapter/integration/test_integration.py
index 42ff8b881f..ccea105bcf 100644
--- a/tests/core/engine_adapter/integration/test_integration.py
+++ b/tests/core/engine_adapter/integration/test_integration.py
@@ -2630,6 +2630,10 @@ def _mutate_config(current_gateway_name: str, config: Config):
     assert context.default_dialect == "duckdb"
 
     schema = ctx.schema(TEST_SCHEMA)
+    seed_columns_to_types = {
+        "item_id": exp.DataType.build("integer"),
+        "event_date": exp.DataType.build("date"),
+    }
     seed_query = ctx.input_data(
         pd.DataFrame(
             [
@@ -2643,13 +2647,15 @@ def _mutate_config(current_gateway_name: str, config: Config):
             ],
             columns=["item_id", "event_date"],
         ),
-        columns_to_types={
-            "item_id": exp.DataType.build("integer"),
-            "event_date": exp.DataType.build("date"),
-        },
+        columns_to_types=seed_columns_to_types,
     )
     context.upsert_model(
-        create_sql_model(name=f"{schema}.seed_model", query=seed_query, kind="FULL")
+        create_sql_model(
+            name=f"{schema}.seed_model",
+            query=seed_query,
+            kind="FULL",
+            columns=seed_columns_to_types,
+        )
     )
 
     table_format = ""