diff --git a/tests/Integration/UpdateCompanyUserTest.php b/tests/Integration/UpdateCompanyUserTest.php
index 2784da2a5755..1fb493fb7feb 100644
--- a/tests/Integration/UpdateCompanyUserTest.php
+++ b/tests/Integration/UpdateCompanyUserTest.php
@@ -101,11 +101,12 @@ class UpdateCompanyUserTest extends TestCase
                         ->first();
 
         $this->assertNotNull($company_user);
-        
+
         $company_user->settings = $settings;
 
         $this->user->company_user = $company_user;
-
+        $this->user->setRelation('company_user', $company_user);
+        
         $user['company_user'] = $company_user->toArray();
 
         $response = null;