diff --git a/wallet/tests/deploy.rs b/wallet/tests/deploy.rs
index e6e9c268b9..2d517698e9 100644
--- a/wallet/tests/deploy.rs
+++ b/wallet/tests/deploy.rs
@@ -6,6 +6,7 @@ use solana::fullnode::Fullnode;
 use solana::leader_scheduler::LeaderScheduler;
 use solana::mint::Mint;
 use solana::rpc_request::{RpcClient, RpcRequest, RpcRequestHandler};
+use solana::storage_stage::STORAGE_ROTATE_TEST_COUNT;
 use solana::vote_signer_proxy::VoteSignerProxy;
 use solana_drone::drone::run_local_drone;
 use solana_sdk::bpf_loader;
@@ -55,6 +56,7 @@ fn test_wallet_deploy_program() {
         &ledger_path,
         false,
         None,
+        STORAGE_ROTATE_TEST_COUNT,
     );
 
     let (sender, receiver) = channel();