diff --git a/tests/integration/persist_constraints/models/schema.yml b/tests/integration/persist_constraints/models/schema.yml
index d21e28b2..9a6bf88d 100644
--- a/tests/integration/persist_constraints/models/schema.yml
+++ b/tests/integration/persist_constraints/models/schema.yml
@@ -52,4 +52,11 @@ models:
   - name: table_model_contract
     config:
       contract:
-        enforced: true
\ No newline at end of file
+        enforced: true
+    columns:
+      - name: id
+        data_type: int
+      - name: name
+        data_type: string
+      - name: date
+        data_type: date
\ No newline at end of file