Update tests to use AnimationController