diff --git a/tests/integration/027_custom_flow/main.tf b/tests/integration/027_custom_flow/main.tf
index afc8d06e..83893bf4 100644
--- a/tests/integration/027_custom_flow/main.tf
+++ b/tests/integration/027_custom_flow/main.tf
@@ -23,11 +23,11 @@ resource "env0_custom_flow" "test" {
 }
 
 data "env0_custom_flow" "test" {
-  name = env0_custom_flow.test.name
+  name       = env0_custom_flow.test.name
+  depends_on = [env0_custom_flow.test]
 }
 
 resource "env0_custom_flow_assignment" "assignment" {
-  depends_on  = [env0_custom_flow.test]
   scope_id    = env0_project.project.id
   template_id = data.env0_custom_flow.test.id
 }