diff --git a/public/script.js b/public/script.js
index 75b6b9949..013d5766c 100644
--- a/public/script.js
+++ b/public/script.js
@@ -1138,8 +1138,9 @@ function isStreamingEnabled() {
 class StreamingProcessor {
     onStartStreaming(text) {
         saveReply(this.type, text);
+        const messageId = count_view_mes - 1;
         hideSwipeButtons();
-        return (count_view_mes - 1);
+        return messageId;
     }
 
     onProgressStreaming(messageId, text) {
diff --git a/readme.md b/readme.md
index 9cc19da86..e15514544 100644
--- a/readme.md
+++ b/readme.md
@@ -144,6 +144,7 @@ Contact us on Discord: Cohee#1207 or RossAscends#1779
 * Cohee's TAI mod: Public domain
 * RossAscends' additions: Public domain
 * Portions of CncAnon's TavernAITurbo mod: Unknown license
+* Waifu mode inspired by the work of PepperTaco (https://github.com/peppertaco/Tavern/)
 * Thanks Pygmalion University for being awesome testers and suggesting cool features! 
 * Thanks oobabooga for compiling presets for TextGen
 * poe-api client adapted from https://github.com/ading2210/poe-api (GPL v3)