Forgot to add the transition stage enumeration