diff --git a/lib/inner_drawer.dart b/lib/inner_drawer.dart
index a2cb9e2..482e3e3 100644
--- a/lib/inner_drawer.dart
+++ b/lib/inner_drawer.dart
@@ -252,16 +252,11 @@ class InnerDrawerState extends State<InnerDrawer>
 
   /// get width of screen after initState
   void _updateWidth() {
-    WidgetsBinding.instance!.addPostFrameCallback((_) {
-      final RenderBox? box =
-          _drawerKey.currentContext!.findRenderObject() as RenderBox?;
-      //final RenderBox box = context.findRenderObject();
-      if (box != null &&
-          box.hasSize &&
-          box.size != null &&
-          box.size.width > 300)
+    WidgetsBinding.instance.addPostFrameCallback((_) {
+      double width = MediaQuery.of(context).size.width;
+      if (width > 250)
         setState(() {
-          _initWidth = box.size.width;
+          _initWidth = width;
         });
     });
   }