diff --git a/src/stubs/tests/feature/full.stub b/src/stubs/tests/feature/full.stub
index 7b74831..fd4d20f 100644
--- a/src/stubs/tests/feature/full.stub
+++ b/src/stubs/tests/feature/full.stub
@@ -88,9 +88,9 @@ class ManageMastersTest extends TestCase
 
         $this->seePageIs(route('masters.show', $singleMstr));
 
-        $this->seeInDatabase('masters', [
+        $this->seeInDatabase('masters', $this->getEditFields([
             'id' => $singleMstr->id,
-        ] + $this->getEditFields());
+        ]));
     }
 
     /** @test */
diff --git a/tests/Generators/FeatureTestGeneratorTest.php b/tests/Generators/FeatureTestGeneratorTest.php
index a071491..0df88a0 100644
--- a/tests/Generators/FeatureTestGeneratorTest.php
+++ b/tests/Generators/FeatureTestGeneratorTest.php
@@ -144,9 +144,9 @@ class Manage{$this->plural_model_name}Test extends TestCase
 
         \$this->seePageIs(route('{$this->table_name}.show', \${$this->single_model_var_name}));
 
-        \$this->seeInDatabase('{$this->table_name}', [
+        \$this->seeInDatabase('{$this->table_name}', \$this->getEditFields([
             'id' => \${$this->single_model_var_name}->id,
-        ] + \$this->getEditFields());
+        ]));
     }
 
     /** @test */
@@ -340,9 +340,9 @@ class Manage{$this->plural_model_name}Test extends TestCase
 
         \$this->seePageIs(route('{$this->table_name}.show', \${$this->single_model_var_name}));
 
-        \$this->seeInDatabase('{$this->table_name}', [
+        \$this->seeInDatabase('{$this->table_name}', \$this->getEditFields([
             'id' => \${$this->single_model_var_name}->id,
-        ] + \$this->getEditFields());
+        ]));
     }
 
     /** @test */