diff --git a/tests/generate_dags_cases/input/basic_dag.yml b/tests/generate_dags_cases/input/basic_dag.yml
index 8e654943..0e80c3d6 100644
--- a/tests/generate_dags_cases/input/basic_dag.yml
+++ b/tests/generate_dags_cases/input/basic_dag.yml
@@ -7,19 +7,15 @@ default_args:
 catchup: false
 custom_callbacks:
   on_success_callback:
-    module: callbacks.slack_messages
-    callable: inform_success
+    module: time
+    callable: sleep
     args:
-      connection_id: DATACOVES_SLACK
-      message: Custom YML success message
-      color: 0000FF
+      secs: 1
   on_failure_callback:
-    module: callbacks.slack_messages
-    callable: inform_failure
+    module: time
+    callable: sleep
     args:
-      connection_id: DATACOVES_SLACK
-      message: Custom YML error message
-      color: 9900FF
+      secs: 1
 nodes:
   transform:
     type: task
diff --git a/tox.ini b/tox.ini
index 2a9b5c77..922ee0f0 100644
--- a/tox.ini
+++ b/tox.ini
@@ -22,10 +22,10 @@ allowlist_externals =
     echo
     pytest
 commands =
-    poetry add apache-airflow
+    poetry add apache-airflow kubernetes
     poetry run pytest --cov-report=xml:coverage-reports/dbt110-coverage.xml --cov-report term-missing  --cov=dbt_coves tests/
     coverage report -m
-    poetry remove apache-airflow
+    poetry remove apache-airflow kubernetes
 
 [testenv:dbt160]
 passenv = *
@@ -42,10 +42,10 @@ allowlist_externals =
     echo
     pytest
 commands =
-    poetry add apache-airflow
+    poetry add apache-airflow kubernetes
     poetry run pytest --cov-report=xml:coverage-reports/coverage.xml --cov-report term-missing  --cov=dbt_coves tests/
     coverage report -m
-    poetry remove apache-airflow
+    poetry remove apache-airflow kubernetes
 
 [gh-actions]
 python =