From 5670e924a5820c3860496a5aa3cdc7cb40369b72 Mon Sep 17 00:00:00 2001
From: SillyLossy <titan0599@gmail.com>
Date: Sun, 7 May 2023 16:31:09 +0300
Subject: [PATCH] Fix in-context messages line for regens

---
 public/script.js | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/public/script.js b/public/script.js
index a64473b41..aaea89823 100644
--- a/public/script.js
+++ b/public/script.js
@@ -2324,9 +2324,9 @@ async function Generate(type, { automatic_trigger, force_name2, resolve, reject,
 } //generate ends
 
 function setInContextMessages(lastmsg, type) {
-    $("#chat").children().removeClass('lastInContext');
+    $("#chat .mes").removeClass('lastInContext');
 
-    if (type === 'swipe') {
+    if (type === 'swipe' || type === 'regenerate') {
         lastmsg++;
     }