diff --git a/tests/Integration/UpdateCompanyUserTest.php b/tests/Integration/UpdateCompanyUserTest.php
index 326a4aeeea88..30d0cb36fb51 100644
--- a/tests/Integration/UpdateCompanyUserTest.php
+++ b/tests/Integration/UpdateCompanyUserTest.php
@@ -105,8 +105,8 @@ class UpdateCompanyUserTest extends TestCase
         $company_user->settings = $settings;
 
         // $this->user->company_user = $company_user;
-        // $this->user->setRelation('company_user', $company_user);
-
+        $this->user->setRelation('company_user', $company_user);
+        $user = $this->user->toArray();
         $user['company_user'] = $company_user->toArray();
 
         $response = null;