diff --git a/test/features/composability/scenarios/forwarder_contract_upgrade.scen.json b/test/features/composability/scenarios/forwarder_contract_upgrade.scen.json
index 2634f2c7f..029d9726b 100644
--- a/test/features/composability/scenarios/forwarder_contract_upgrade.scen.json
+++ b/test/features/composability/scenarios/forwarder_contract_upgrade.scen.json
@@ -8,6 +8,9 @@
                 "sc:forwarder": {
                     "code": "file:../forwarder/output/forwarder.wasm"
                 },
+                "sc:reference": {
+                    "code": "file:../vault/output/vault-upgrade.wasm"
+                },
                 "sc:child": {
                     "code": "file:../vault/output/vault.wasm",
                     "owner": "sc:forwarder"
@@ -16,14 +19,14 @@
         },
         {
             "step": "scCall",
-            "id": "upgrade-vault-to-forwarder",
+            "id": "upgrade-vault-from-source",
             "tx": {
                 "from": "address:a_user",
                 "to": "sc:forwarder",
                 "function": "upgrade_vault_from_source",
                 "arguments": [
                     "sc:child",
-                    "sc:child"
+                    "sc:reference"
                 ],
                 "gasLimit": "500,000,000",
                 "gasPrice": "0"
@@ -37,7 +40,7 @@
             "step": "checkState",
             "accounts": {
                 "sc:child": {
-                    "code": "file:../vault/output/vault.wasm"
+                    "code": "file:../vault/output/vault-upgrade.wasm"
                 },
                 "+": ""
             }
diff --git a/test/features/composability/vault/output/vault-promises.wasm b/test/features/composability/vault/output/vault-promises.wasm
index bb2688fae..89da5d0a1 100755
Binary files a/test/features/composability/vault/output/vault-promises.wasm and b/test/features/composability/vault/output/vault-promises.wasm differ
diff --git a/test/features/composability/vault/output/vault-upgrade.wasm b/test/features/composability/vault/output/vault-upgrade.wasm
new file mode 100755
index 000000000..1b8c1d642
Binary files /dev/null and b/test/features/composability/vault/output/vault-upgrade.wasm differ
diff --git a/test/features/composability/vault/output/vault.wasm b/test/features/composability/vault/output/vault.wasm
index 18b021b61..71ddc12a0 100755
Binary files a/test/features/composability/vault/output/vault.wasm and b/test/features/composability/vault/output/vault.wasm differ