diff --git a/integration_tests/macros/operations/create_source_table.sql b/integration_tests/macros/operations/create_source_table.sql
index 6794e51..f17f3fd 100644
--- a/integration_tests/macros/operations/create_source_table.sql
+++ b/integration_tests/macros/operations/create_source_table.sql
@@ -9,7 +9,7 @@ reset enable_case_sensitive_identifier;
 
 {% set target_schema=api.Relation.create(
     database=target.database,
-    schema="codegen_integration_tests__data_source_schema"
+    schema="{{ schema }}__data_source_schema"
 ) %}
 
 
diff --git a/integration_tests/tests/test_generate_model_struct_yaml.sql b/integration_tests/tests/test_generate_model_struct_yaml.sql
index 148323b..48378d6 100644
--- a/integration_tests/tests/test_generate_model_struct_yaml.sql
+++ b/integration_tests/tests/test_generate_model_struct_yaml.sql
@@ -3,7 +3,7 @@
 -- test all args
 {% set actual_source_yaml = codegen.generate_source(
     database_name=target.database,
-    schema_name='codegen_integration_tests__data_source_schema',
+    schema_name='{{ schema }}__data_source_schema',
     table_names=['codegen_integration_tests__data_source_table_nested_array'],
     generate_columns=True,
     include_descriptions=True