Unit tests for store, small fix for removing initial state