Cleaned up TransitionState imports.