fix more tests

tserg · Nov 24, 2024 · a16d67e · a16d67e
1 parent c552876
commit a16d67e
Showing 1 changed file with 2 additions and 6 deletions.
diff --git 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": {"*": ["*"]}},
     }