diff --git a/src/Hydrate/EntityHydrator.php b/src/Hydrate/EntityHydrator.php
index 77d5096d..5e9f9da9 100644
--- a/src/Hydrate/EntityHydrator.php
+++ b/src/Hydrate/EntityHydrator.php
@@ -47,6 +47,8 @@ public function schema(string $entity, Context $context): Schema
                 throw new Exception('Schema for entity: ' . $entity . ' not found');
             }
         }
+        
+        $this->cacheSchema[$entity] = $schema;
 
         return $schema;
     }