diff --git a/tests/src/migration.rs b/tests/src/migration.rs
index eb585f9..6a11ba8 100644
--- a/tests/src/migration.rs
+++ b/tests/src/migration.rs
@@ -1,13 +1,13 @@
-use casper_engine_test_support::DEFAULT_ACCOUNT_ADDR;
-use casper_types::runtime_args;
+use casper_engine_test_support::{ExecuteRequestBuilder, DEFAULT_ACCOUNT_ADDR};
+use casper_types::{runtime_args, ContractHash, RuntimeArgs, U256};
 
 use crate::utility::{
     constants::{
         ARG_DECIMALS, ARG_NAME, ARG_SYMBOL, ARG_TOTAL_SUPPLY, CEP18_CONTRACT_WASM,
-        CEP18_TOKEN_CONTRACT_KEY, DECIMALS_KEY, NAME_KEY, SYMBOL_KEY, TOKEN_DECIMALS, TOKEN_NAME,
-        TOKEN_SYMBOL, TOKEN_TOTAL_SUPPLY, TOTAL_SUPPLY_KEY,
+        TOKEN_DECIMALS, TOKEN_NAME,
+        TOKEN_SYMBOL, TOKEN_TOTAL_SUPPLY, 
     },
-    installer_request_builders::setup,
+    installer_request_builders::{setup, TestContext},
 };
 
 #[test]
@@ -17,7 +17,7 @@ fn should_have_queryable_properties() {
         .get_account(*DEFAULT_ACCOUNT_ADDR)
         .expect("should have account");
 
-    let version_0 = account
+    let version_0 = pre_account
         .named_keys()
         .get("cep18_contract_version_CasperTest")
         .and_then(|key| key.into_hash())
@@ -36,11 +36,13 @@ fn should_have_queryable_properties() {
     )
     .build();
 
+    builder.exec(install_request_1).expect_success().commit();
+
     let post_account = builder
         .get_account(*DEFAULT_ACCOUNT_ADDR)
         .expect("should have account");
 
-    let version_1 = account
+    let version_1 = post_account
         .named_keys()
         .get("cep18_contract_version_CasperTest")
         .and_then(|key| key.into_hash())