diff --git a/tests/test_integration.py b/tests/test_integration.py
index d673af0f..29c9ed4c 100644
--- a/tests/test_integration.py
+++ b/tests/test_integration.py
@@ -232,13 +232,10 @@ def test_auxilliary_cli(tmp_path, mgf_small, monkeypatch):
     with pytest.raises(FileExistsError):
         run(["configure", "-o", "test.yaml"])
 
-    with open("casanovo.yaml") as f_in:
+    with open("casanovo.yaml") as f_in, open("small.yaml", "w") as f_out:
         config = yaml.safe_load(f_in)
-
-    config["max_epochs"] = 1
-    config["n_layers"] = 1
-
-    with open("small.yaml", "w") as f_out:
+        config["max_epochs"] = 1
+        config["n_layers"] = 1
         yaml.dump(config, f_out)
 
     train_args = [