add in some missing state transitions