diff --git a/migrator/accounts2_it_test.go b/migrator/accounts2_it_test.go
index eae4bd94e..540fd5c32 100755
--- a/migrator/accounts2_it_test.go
+++ b/migrator/accounts2_it_test.go
@@ -121,10 +121,10 @@ func testAcc2ITFlush(t *testing.T) {
 	if acc2Migrator.dmOut.DataManager().DataDB().GetStorageType() != utils.REDIS {
 		t.Errorf("Unexpected datadb type : %+v", acc2Migrator.dmOut.DataManager().DataDB().GetStorageType())
 	}
-	if acc2Migrator.storDBIn.StorDB().GetStorageType() != utils.MAPSTOR {
+	if acc2Migrator.storDBIn.StorDB().GetStorageType() != utils.INTERNAL {
 		t.Errorf("Unexpected datadb type : %+v", acc2Migrator.storDBIn.StorDB().GetStorageType())
 	}
-	if acc2Migrator.storDBOut.StorDB().GetStorageType() != utils.MAPSTOR {
+	if acc2Migrator.storDBOut.StorDB().GetStorageType() != utils.INTERNAL {
 		t.Errorf("Unexpected datadb type : %+v", acc2Migrator.storDBOut.StorDB().GetStorageType())
 	}
 }
diff --git a/migrator/storage_map_stordb.go b/migrator/storage_map_stordb.go
index ce52b3bd3..4715a6a83 100755
--- a/migrator/storage_map_stordb.go
+++ b/migrator/storage_map_stordb.go
@@ -26,13 +26,13 @@ import (
 func newMapStorDBMigrator(stor engine.StorDB) (mpMig *mapStorDBMigrator) {
 	return &mapStorDBMigrator{
 		storDB: &stor,
-		mp:     stor.(*engine.MapStorage),
+		iDB:     stor.(*engine.InternalDB),
 	}
 }
 
 type mapStorDBMigrator struct {
 	storDB   *engine.StorDB
-	mp       *engine.MapStorage
+	iDB       *engine.InternalDB
 	dataKeys []string
 	qryIdx   *int
 }