diff --git a/.github/workflows/CI.yaml b/.github/workflows/CI.yaml
index aad6175..a483e2e 100644
--- a/.github/workflows/CI.yaml
+++ b/.github/workflows/CI.yaml
@@ -79,10 +79,11 @@ jobs:
           cd tests_with_diesel_cli
 
           # with default schema
+          diesel setup
           diesel migration run
           cargo test
-          rm -rf src/schema.rs
           diesel migration revert
+          rm src/schema.rs diesel.toml
 
           # create a custom schema
           diesel migration run --config-file custom.diesel.toml