Update WifiStateMachine.java

This reverts commit b301684
CRs-fixed: 509553