diff --git a/tests/test_b2shareoperator.py b/tests/test_b2shareoperator.py
index 702e0875a9418389835d5b72ef34e92ddced5fe3..d167bc281c98f538881069fd2c90b57e5c3febc0 100644
--- a/tests/test_b2shareoperator.py
+++ b/tests/test_b2shareoperator.py
@@ -4,13 +4,15 @@ from airflow.utils.dates import days_ago
 from dags.b2shareoperator import B2ShareOperator
 from airflow import DAG
 from airflow.models.taskinstance import TaskInstance
+from tests.test_utils.config import conf_vars
+
 
 DEFAULT_DATE = '2019-10-03'
 TEST_DAG_ID = 'test_my_custom_operator'
 
 class B2ShareOperatorTest(unittest.TestCase):
    def setUp(self):
-       self.dag = DAG(TEST_DAG_ID, schedule_interval='@daily', default_args={'start_date' : days_ago(2)})
+       self.dag = DAG(TEST_DAG_ID, schedule_interval='@daily', default_args={'start_date' : days_ago(2)}, params={"oid": "111"})
        self.op = B2ShareOperator(
            dag=self.dag,
            task_id='test',
@@ -19,8 +21,8 @@ class B2ShareOperatorTest(unittest.TestCase):
        self.ti = TaskInstance(task=self.op, execution_date=days_ago(1))
 
    def test_execute_no_trigger(self):
-       
-       self.ti.run(ignore_ti_state=False, context={'params': {'oid': 111}})
+       #with conf_vars({('params': {'oid': 1111})}):
+       self.ti.run(ignore_ti_state=False)
        print(self.ti.state)
        self.assertEqual(State.SUCCESS, self.ti.state)
        # Assert something related to tasks results
\ No newline at end of file