diff --git a/tests/end-to-end/test_database.py b/tests/end-to-end/test_database.py
index 9209ec39..f9c36214 100644
--- a/tests/end-to-end/test_database.py
+++ b/tests/end-to-end/test_database.py
@@ -25,7 +25,7 @@ def write_configuration(
 source_directories:
     - {source_directory}
 repositories:
-    - {repository_path}
+    - path: {repository_path}
 borgmatic_source_directory: {borgmatic_source_directory}
 
 encryption_passphrase: "test"
@@ -96,7 +96,7 @@ def write_custom_restore_configuration(
 source_directories:
     - {source_directory}
 repositories:
-    - {repository_path}
+    - path: {repository_path}
 borgmatic_source_directory: {borgmatic_source_directory}
 
 encryption_passphrase: "test"
@@ -158,7 +158,7 @@ def write_simple_custom_restore_configuration(
 source_directories:
     - {source_directory}
 repositories:
-    - {repository_path}
+    - path: {repository_path}
 borgmatic_source_directory: {borgmatic_source_directory}
 
 encryption_passphrase: "test"
diff --git a/tests/integration/config/test_generate.py b/tests/integration/config/test_generate.py
index 5df1825a..e332b899 100644
--- a/tests/integration/config/test_generate.py
+++ b/tests/integration/config/test_generate.py
@@ -58,14 +58,13 @@ foo:
         - baz
         - quux
 
-location:
-    repositories:
-        - one
-        - two
+repositories:
+    - one
+    - two
 
-    # This comment should be kept.
-    # COMMENT_OUT
-    other: thing
+# This comment should be kept.
+# COMMENT_OUT
+other: thing
     '''
 
     # flake8: noqa
@@ -75,13 +74,12 @@ location:
 #         - baz
 #         - quux
 
-location:
-    repositories:
-        - one
-        - two
+repositories:
+    - one
+    - two
 
-    # This comment should be kept.
-#     other: thing
+# This comment should be kept.
+# other: thing
     '''
 
     assert module.comment_out_optional_configuration(config.strip()) == expected_config.strip()