diff --git a/packages/schemas/resources/views/components/flex.blade.php b/packages/schemas/resources/views/components/flex.blade.php
index 9007c94571..91f6c15051 100644
--- a/packages/schemas/resources/views/components/flex.blade.php
+++ b/packages/schemas/resources/views/components/flex.blade.php
@@ -4,6 +4,8 @@
     use Filament\Schemas\Components\Component;
     use Filament\Support\Enums\VerticalAlignment;
 
+    $statePath = $getStatePath();
+
     $fromBreakpoint = $getFromBreakpoint();
     $verticalAlignment = $getVerticalAlignment();