Fix import state for state tests

Storage was overwritten