Remove WaitRunning method from state