From fa7ce1bc24563512c3d978f1b0dda60dbf9c16a4 Mon Sep 17 00:00:00 2001
From: David Bomba <turbo124@gmail.com>
Date: Tue, 3 Sep 2024 09:49:29 +1000
Subject: [PATCH] Fixes for tests

---
 tests/Integration/UpdateCompanyUserTest.php | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

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;