Convert to use InMemoryStore