diff --git a/tests/integration_tests/modules/test_disk_setup.py b/tests/integration_tests/modules/test_disk_setup.py
index bc599741262..97be1abe2f2 100644
--- a/tests/integration_tests/modules/test_disk_setup.py
+++ b/tests/integration_tests/modules/test_disk_setup.py
@@ -191,7 +191,7 @@ def test_disk_setup_when_mounted(
             UPDATED_PARTPROBE_USERDATA,
         )
         client.execute(
-            "sed -i 's/write_files/write_files\\n -  mounts/' "
+            "sed -i 's/write_files$/write_files\\n  - mounts/' "
             "/etc/cloud/cloud.cfg"
         )