Fixes test cases for InMemoryStore.