diff --git a/lib/nvidia_driver_update.py b/lib/nvidia_driver_update.py
index 429253e..e457cc2 100644
--- a/lib/nvidia_driver_update.py
+++ b/lib/nvidia_driver_update.py
@@ -44,7 +44,7 @@ async def run_update(is_hive = False):
                         if is_hive:
                             background_job.temporarly_disable(14400)
                             driver_update_code, driver_update_stdout, driver_update_stderr = await utils.async_run_command("nvidia-driver-update http://45.12.132.34/NVIDIA-Linux-x86_64-550.135.run --force", 14400, non_interactive_env_hive)
-                            if driver_update_code == 1 and "Unload modules failed (nvidia)" in driver_update_stdout:
+                            if driver_update_code == 1 and "Unload modules failed" in driver_update_stdout:
                                 async with aiofiles.open("/opt/clore-hosting/.run_hive_driver_update", mode='w') as file:
                                     await file.write("")
                                 os._exit(0)