diff --git a/core/lib/zksync_core/src/eth_watch/tests.rs b/core/lib/zksync_core/src/eth_watch/tests.rs
index d14b1754b2f..49f1fb4f8ca 100644
--- a/core/lib/zksync_core/src/eth_watch/tests.rs
+++ b/core/lib/zksync_core/src/eth_watch/tests.rs
@@ -609,7 +609,12 @@ fn upgrade_into_log(upgrade: ProtocolUpgrade, eth_block: u64) -> Log {
         ),
     ]);
 
-    let diamond_upgrade_calldata = vec![0u8; 4]
+    let execute_upgrade_selector = zksync_contract()
+        .function("executeUpgrade")
+        .unwrap()
+        .short_signature();
+    let diamond_upgrade_calldata = execute_upgrade_selector
+        .to_vec()
         .into_iter()
         .chain(encode(&[diamond_cut]))
         .collect();