diff --git a/flutter_filament_federated/flutter_filament/example/lib/main.dart b/flutter_filament_federated/flutter_filament/example/lib/main.dart
index 9f709946..925ff227 100644
--- a/flutter_filament_federated/flutter_filament/example/lib/main.dart
+++ b/flutter_filament_federated/flutter_filament/example/lib/main.dart
@@ -110,7 +110,7 @@ class ExampleWidgetState extends State<ExampleWidget> {
             if (isInitialized)
               Positioned.fill(
                 child: ExampleViewport(
-                    controller: _plugin,
+                    controller: isInitialized ? _plugin : null,
                     entityTransformController: _transformController,
                     padding: _viewportMargin,
                     keyboardFocusNode: _sharedFocusNode),