diff --git a/tests/unit/cli/storage_layout/test_storage_layout_overrides.py b/tests/unit/cli/storage_layout/test_storage_layout_overrides.py
index d7db30cc80..bf92754666 100644
--- a/tests/unit/cli/storage_layout/test_storage_layout_overrides.py
+++ b/tests/unit/cli/storage_layout/test_storage_layout_overrides.py
@@ -41,9 +41,7 @@ def test_storage_layout_overrides_json():
     input_json = {
         "language": "Vyper",
         "sources": {"contracts/foo.vy": {"content": code}},
-        "storage_layout_overrides": {
-            "contracts/foo.vy": {"content": {"storage_layout": storage_layout_overrides}}
-        },
+        "storage_layout_overrides": {"contracts/foo.vy": {"content": storage_layout_overrides}},
         "settings": {"outputSelection": {"*": ["*"]}},
     }
 
@@ -68,9 +66,7 @@ def test_storage_layout_overrides_output():
     input_json = {
         "language": "Vyper",
         "sources": {"contracts/foo.vy": {"content": code}},
-        "storage_layout_overrides": {
-            "contracts/foo.vy": {"content": {"storage_layout": storage_layout_overrides}}
-        },
+        "storage_layout_overrides": {"contracts/foo.vy": {"content": storage_layout_overrides}},
         "settings": {"outputSelection": {"*": ["*"]}},
     }