Merge pull request #8 from formapro/fix-builder-hook

Fix Object Builder Hook
formapro · Oct 4, 2017 · 244edd2 · 244edd2
2 parents 730e2b5 + 739a581
commit 244edd2
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/ObjectBuilderHook.php b/src/ObjectBuilderHook.php
@@ -6,7 +6,7 @@
 class ObjectBuilderHook
 {
     /**
-     * @var PvmClassMap
+     * @var array
      */
     private $classMap;
 
@@ -15,14 +15,14 @@ class ObjectBuilderHook
      */
     public function __construct(array $classMap = [])
     {
-        $this->classMap = new PvmClassMap($classMap);
+        $this->classMap = (new PvmClassMap($classMap))->get();
     }
 
     public function register()
     {
         register_global_hook('get_object_class', function(array $values) {
             if (isset($values['schema'])) {
-                if (false == array_key_exists($values['schema'], $this->classMap->get())) {
+                if (false == array_key_exists($values['schema'], $this->classMap)) {
                     throw new \LogicException(sprintf('An object has class set "%s" but there is no class for it', $values['class']));
                 }