diff --git a/lib/src/widgets/sliver_aligned_grid.dart b/lib/src/widgets/sliver_aligned_grid.dart
index 0c799d5..10eb18a 100644
--- a/lib/src/widgets/sliver_aligned_grid.dart
+++ b/lib/src/widgets/sliver_aligned_grid.dart
@@ -188,7 +188,7 @@ class _Gap extends StatelessWidget {
 
   @override
   Widget build(BuildContext context) {
-    final axis = axisDirectionToAxis(Scrollable.of(context)!.axisDirection);
+    final axis = axisDirectionToAxis(Scrollable.of(context).axisDirection);
     return axis == Axis.vertical
         ? SizedBox(height: mainAxisExtent)
         : SizedBox(width: mainAxisExtent);
diff --git a/lib/src/widgets/staggered_grid.dart b/lib/src/widgets/staggered_grid.dart
index 796b053..f788294 100644
--- a/lib/src/widgets/staggered_grid.dart
+++ b/lib/src/widgets/staggered_grid.dart
@@ -82,7 +82,7 @@ class StaggeredGrid extends MultiChildRenderObjectWidget {
       mainAxisSpacing: mainAxisSpacing,
       crossAxisSpacing: crossAxisSpacing,
       axisDirection: axisDirection ??
-          Scrollable.of(context)?.axisDirection ??
+          Scrollable.maybeOf(context)?.axisDirection ??
           AxisDirection.down,
       textDirection: Directionality.of(context),
     );
@@ -98,7 +98,7 @@ class StaggeredGrid extends MultiChildRenderObjectWidget {
       ..mainAxisSpacing = mainAxisSpacing
       ..crossAxisSpacing = crossAxisSpacing
       ..axisDirection = axisDirection ??
-          Scrollable.of(context)?.axisDirection ??
+          Scrollable.maybeOf(context)?.axisDirection ??
           AxisDirection.down
       ..textDirection = Directionality.of(context);
   }