diff --git a/tests/Controller/ApiControllerTest.php b/tests/Controller/ApiControllerTest.php
index c63944d..2dea3c7 100755
--- a/tests/Controller/ApiControllerTest.php
+++ b/tests/Controller/ApiControllerTest.php
@@ -66,8 +66,8 @@ final class ControllerTest extends \PHPUnit\Framework\TestCase
         TestUtils::setMember($account, 'id', 1);
 
         $permission = new AccountPermission();
-        $permission->setUnit(1);
-        $permission->setApp(2);
+        $permission->unit = 1;
+        $permission->app  = 2;
         $permission->setPermission(
             PermissionType::READ
             | PermissionType::CREATE
diff --git a/tests/Controller/BackendControllerTest.php b/tests/Controller/BackendControllerTest.php
index 0fea564..aae19ef 100755
--- a/tests/Controller/BackendControllerTest.php
+++ b/tests/Controller/BackendControllerTest.php
@@ -58,8 +58,8 @@ final class BackendControllerTest extends \PHPUnit\Framework\TestCase
         TestUtils::setMember($account, 'id', 1);
 
         $permission = new AccountPermission();
-        $permission->setUnit(1);
-        $permission->setApp(2);
+        $permission->unit = 1;
+        $permission->app  = 2;
         $permission->setPermission(
             PermissionType::READ
             | PermissionType::CREATE