diff --git a/src/Generator/GenerateModifiers.php b/src/Generator/GenerateModifiers.php
index cb95a70..607b9e0 100644
--- a/src/Generator/GenerateModifiers.php
+++ b/src/Generator/GenerateModifiers.php
@@ -28,7 +28,7 @@ public function run(Registry $registry): Registry
     protected function register(Registry $registry, Entity $entity): void
     {
         $role = $entity->getRole();
-
+        assert($role !== null);
         foreach ($entity->getSchemaModifiers() as $modifier) {
             \assert($modifier instanceof SchemaModifierInterface);
             try {
diff --git a/src/Generator/RenderModifiers.php b/src/Generator/RenderModifiers.php
index a13d4fe..e55c657 100644
--- a/src/Generator/RenderModifiers.php
+++ b/src/Generator/RenderModifiers.php
@@ -28,7 +28,7 @@ public function run(Registry $registry): Registry
     protected function register(Registry $registry, Entity $entity): void
     {
         $role = $entity->getRole();
-
+        assert($role !== null);
         foreach ($entity->getSchemaModifiers() as $modifier) {
             \assert($modifier instanceof SchemaModifierInterface);
             try {