diff --git a/tests_e2e/tests/agent_update/rsm_update.py b/tests_e2e/tests/agent_update/rsm_update.py
index cfa1a7d18..05112fc19 100644
--- a/tests_e2e/tests/agent_update/rsm_update.py
+++ b/tests_e2e/tests/agent_update/rsm_update.py
@@ -114,7 +114,7 @@ def _prepare_agent(self) -> None:
 
     @staticmethod
     def _verify_agent_update_flag_enabled(vm: VirtualMachineClient) -> bool:
-        result: VirtualMachine = vm.get_description()
+        result: VirtualMachine = vm.get_model()
         flag: bool = result.os_profile.linux_configuration.enable_vm_agent_platform_updates
         if flag is None:
             return False