diff --git a/tests/test_integration/test_data_testing.py b/tests/test_integration/test_data_testing.py
index cad50a1..cad3b1e 100644
--- a/tests/test_integration/test_data_testing.py
+++ b/tests/test_integration/test_data_testing.py
@@ -40,7 +40,7 @@ def test_validation_on_mysql_succeeds():
         ),
     ]
 
-    validation_result = run_data_test_on_mysql(
+    validation_result = run_data_test_on_mysql.run(
         name="lolafect-testing-test_validation_on_mysql_succeeds",
         mysql_credentials=TEST_LOLACONFIG.DW_CREDENTIALS,
         query=test_query,
@@ -73,7 +73,7 @@ def test_validation_on_mysql_fails():
         ),
     ]
 
-    validation_result = run_data_test_on_mysql(
+    validation_result = run_data_test_on_mysql.run(
         name="lolafect-testing-test_validation_on_mysql_fails",
         mysql_credentials=TEST_LOLACONFIG.DW_CREDENTIALS,
         query=test_query,