@xiangli-cmu Why is there a `stateMutex`? What's the race condition that's trying to be avoided? It seems like a single lock would be fine.