Make changes to Concurrency Test to account for InMemoryShardStateIO