--- a/tests/src/util/frankenstein.ts
+++ b/tests/src/util/frankenstein.ts
@@ -239,7 +239,7 @@
     }, network.relay[0].wsUri);
 
     // Wait for the next epoch so that the parachains will start cooperating with the relay
-    if (relayUpgradeCompleted) {
+    if (relayUpgradeCompleted && NEW_RELAY_WASM) {
       console.log('\n🚥 Starting timeout for the next epoch before upgrading the parachains code...');
       await waitWithTimer(relayInfo.epochTime);
     }