diff --git a/lib/src/views/camera_to_share/share_image_editor_view.dart b/lib/src/views/camera_to_share/share_image_editor_view.dart
index 5491480..aa3e5a3 100644
--- a/lib/src/views/camera_to_share/share_image_editor_view.dart
+++ b/lib/src/views/camera_to_share/share_image_editor_view.dart
@@ -189,10 +189,12 @@ class _ShareImageEditorView extends State<ShareImageEditorView> {
       for (var x in layers) {
         x.showCustomButtons = false;
       }
+      setState(() {});
       image = await screenshotController.capture(pixelRatio: pixelRatio);
       for (var x in layers) {
         x.showCustomButtons = true;
       }
+      setState(() {});
     } else if (layers.length == 1) {
       if (layers.first is BackgroundLayerData) {
         image = (layers.first as BackgroundLayerData).image.bytes;