diff --git a/tests/test_b2shareoperator.py b/tests/test_b2shareoperator.py
index 118b8afa4490b0760904c3af6d7d876e0649ed25..e5de98e4c18d3b99d1a4ce574fe485a48056e4bc 100644
--- a/tests/test_b2shareoperator.py
+++ b/tests/test_b2shareoperator.py
@@ -9,10 +9,11 @@ TEST_DAG_ID = 'test_my_custom_operator'
 
 class B2ShareOperatorTest(unittest.TestCase):
    def setUp(self):
-       self.dag = DAG(TEST_DAG_ID, schedule_interval='@daily',name='testing_dag', default_args={'start_date' : DEFAULT_DATE})
+       self.dag = DAG(TEST_DAG_ID, schedule_interval='@daily', default_args={'start_date' : DEFAULT_DATE})
        self.op = B2ShareOperator(
            dag=self.dag,
            task_id='test',
+           name='test_name'
        )
        self.ti = TaskInstance(task=self.op, execution_date=DEFAULT_DATE)
 
diff --git a/tests/test_dag.py b/tests/test_dag.py
index 08c39ad3af44b5ea0cde234cece6b599436eef13..8a1faa1b7b3deb63adcb7bb56a2342abd72adcdd 100644
--- a/tests/test_dag.py
+++ b/tests/test_dag.py
@@ -10,4 +10,4 @@ class TestADag(unittest.TestCase):
        dag = self.dagbag.get_dag(dag_id='firsto')
        assert self.dagbag.import_errors == {}
        assert dag is not None
-       assert len(dag.tasks) == 4
\ No newline at end of file
+       self.assertEqual(len(dag.tasks), 3, f"Actually: {len(dag.tasks)}")
\ No newline at end of file