Change these and remove the willTransition hook bs!