From 726d0eda432c4ed5e1ac6ebc118ddf02b186d6f4 Mon Sep 17 00:00:00 2001
From: Sander Bosma <sanderbosma@gmail.com>
Date: Tue, 21 Feb 2023 14:48:49 +0100
Subject: [PATCH] override wasm before importing storage

---
 packages/chopsticks/src/setup.ts | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/packages/chopsticks/src/setup.ts b/packages/chopsticks/src/setup.ts
index e66ac2b0..70d8243d 100644
--- a/packages/chopsticks/src/setup.ts
+++ b/packages/chopsticks/src/setup.ts
@@ -76,8 +76,10 @@ export const setup = async (argv: Config) => {
 
   if (argv.timestamp) await timeTravel(chain, argv.timestamp)
 
-  await importStorage(chain, argv['import-storage'])
+  // override wasm before importing storage, in case new pallets have been
+  // added that have storage imports
   await overrideWasm(chain, argv['wasm-override'])
+  await importStorage(chain, argv['import-storage'])
 
   return { chain, api, ws: provider }
 }