fixed the stuck AnimatedButton issue