Work around document constructor

abellion · Sep 15, 2017 · b794d5f · b794d5f
1 parent 70749e7
commit b794d5f
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/src/Document.php b/src/Document.php
@@ -76,15 +76,16 @@ public function has(string $offset)
      */
     public function with(array $fields)
     {
-        $document = [];
+        $document = new static();
+        $document->document = [];
 
         foreach ($fields as $field) {
             if (array_key_exists($field, $this->document)) {
-                $document[$field] = $this->document[$field];
+                $document->document[$field] = $this->document[$field];
             }
         }
 
-        return new static($document);
+        return $document;
     }
 
     /**
@@ -96,15 +97,16 @@ public function with(array $fields)
      */
     public function without(array $fields)
     {
-        $document = [];
+        $document = new static();
+        $document->document = [];
 
         foreach ($this->document as $key => $value) {
             if (!in_array($key, $fields)) {
-                $document[$key] = $this->document[$key];
+                $document->document[$key] = $this->document[$key];
             }
         }
 
-        return new static($document);
+        return $document;
     }