Cleanup CSSTransitionGroup fix a bit