diff --git a/tests/Admin/Form/Element/NamedFormElementTest.php b/tests/Admin/Form/Element/NamedFormElementTest.php
index 9671ce342..3b7d319c7 100644
--- a/tests/Admin/Form/Element/NamedFormElementTest.php
+++ b/tests/Admin/Form/Element/NamedFormElementTest.php
@@ -131,9 +131,10 @@ public function test_add_unique_rule()
         $element->setModel($model);
 
         $model->shouldReceive('getTable')->once()->andReturn('test_table');
+        $model->shouldReceive('getConnectionName')->once()->andReturn('connection');
 
         $element->unique();
-        $this->assertEquals(['key' => ['unique:test_table,key']], $element->getValidationRules());
+        $this->assertEquals(['key' => ['unique:connection.test_table,key']], $element->getValidationRules());
     }
 
     /**