diff --git a/lib/zoom_pinch_overlay.dart b/lib/zoom_pinch_overlay.dart
index 67d1ecc..4b57d27 100644
--- a/lib/zoom_pinch_overlay.dart
+++ b/lib/zoom_pinch_overlay.dart
@@ -111,7 +111,7 @@ class _ZoomOverlayState extends State<ZoomOverlay>
 
     _controllerReset
       ..addListener(() {
-        _transformWidget.currentState!.setMatrix(_animationReset.value);
+        _transformWidget.currentState?.setMatrix(_animationReset.value);
       })
       ..addStatusListener((status) {
         if (status == AnimationStatus.completed) hide();