Fix migration test

aurora-is-near · Jan 26, 2023 · 4021045 · 4021045
1 parent 19f3a12
commit 4021045
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/etc/tests/state-migration-test/src/lib.rs b/etc/tests/state-migration-test/src/lib.rs
@@ -3,7 +3,7 @@
 extern crate alloc;
 
 use alloc::vec::Vec;
-use aurora_engine::engine::{self, EngineState};
+use aurora_engine::state::{self, EngineState};
 use aurora_engine_sdk::near_runtime::Runtime;
 use aurora_engine_sdk::io::{IO, StorageIntermediate};
 use aurora_engine_types::storage;
@@ -18,7 +18,7 @@ struct NewFancyState {
 #[no_mangle]
 pub extern "C" fn state_migration() {
     let mut io = Runtime;
-    let old_state = match engine::get_state(&io) {
+    let old_state = match state::get_state(&io) {
         Ok(state) => state,
         Err(e) => aurora_engine_sdk::panic_utf8(e.as_ref()),
     };