From 7ec3dd39369ff1525927b641181542a724e01f3b Mon Sep 17 00:00:00 2001
From: Who? <155328415+coddrago@users.noreply.github.com>
Date: Sat, 6 Sep 2025 12:36:11 +0700
Subject: [PATCH] Update updater.py

---
 heroku/modules/updater.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/heroku/modules/updater.py b/heroku/modules/updater.py
index 88dfa3a..5c2294b 100644
--- a/heroku/modules/updater.py
+++ b/heroku/modules/updater.py
@@ -315,7 +315,6 @@ class UpdaterMod(loader.Module):
 
         if "LAVHOST" in os.environ:
             await self.client.send_message("lavhostbot", "🔄 Restart")
-            await message.client.disconnect()
             return
 
         await message.client.disconnect()
@@ -417,7 +416,7 @@ class UpdaterMod(loader.Module):
             os.system(f"cd {utils.get_base_dir()} && cd .. && git reset --hard HEAD")
 
         try:
-            if "LAVHOST" in os.environ:
+            if "LAVHOSTtest" in os.environ:
                 msg_obj = await utils.answer(
                     msg_obj,
                     self.strings("lavhost_update").format(