diff --git a/animations/lib/src/basics/06_custom_tween.dart b/animations/lib/src/basics/06_custom_tween.dart
index 8bd56b354..c0c9f10a5 100644
--- a/animations/lib/src/basics/06_custom_tween.dart
+++ b/animations/lib/src/basics/06_custom_tween.dart
@@ -52,9 +52,13 @@ class _CustomTweenDemoState extends State<CustomTweenDemo>
             textColor: Colors.white,
             onPressed: () {
               if (controller.status == AnimationStatus.completed) {
-                controller.reverse();
+                controller.reverse().whenComplete(() {
+                  setState(() {});
+                });
               } else {
-                controller.forward();
+                controller.forward().whenComplete(() {
+                  setState(() {});
+                });
               }
             },
           ),