diff --git a/src/controllers/api/habits.api.controller.tsx b/src/controllers/api/habits.api.controller.tsx
index 894888a..14ad70f 100644
--- a/src/controllers/api/habits.api.controller.tsx
+++ b/src/controllers/api/habits.api.controller.tsx
@@ -46,13 +46,18 @@ export const habitIdApiController = new Hono<{ Variables: AppVariables }>()
       }
 
       return html(
-        <HabitItem
-          item={updatedHabit}
-          x-notification={{
-            type: "success",
-            message: "Habit updated successfully",
-          }}
-        />
+        <>
+          <span
+            x-init={`
+              $notify(${JSON.stringify({
+                type: "success",
+                message: "Habit updated successfully",
+              })});
+              $el.remove();
+            `}
+          />
+          <HabitItem item={updatedHabit} />
+        </>
       );
     }
   )