diff --git a/tests/test_dag.py b/tests/test_dag.py
index 2689dfb9ceedd1fac32ce5ea222580658f6470b3..3a91270f2cc59f8e66c12577c60b767b90ed3c7b 100644
--- a/tests/test_dag.py
+++ b/tests/test_dag.py
@@ -18,4 +18,4 @@ class TestADag(unittest.TestCase):
         dag = self.dagbag.get_dag(dag_id='taskflow_example')
         assert self.dagbag.import_errors == {}
         assert dag is not None
-        self.assertEqual(len(dag.tasks), 3, f"Actually: {len(dag.tasks)}")
+        self.assertEqual(len(dag.tasks), 5, f"Actually: {len(dag.tasks)}")