Updated test to remove the store map