diff --git a/static/koboldai.js b/static/koboldai.js
index 4a8b42d8..e1d98fd1 100644
--- a/static/koboldai.js
+++ b/static/koboldai.js
@@ -803,9 +803,7 @@ function var_changed(data) {
 			image_area.appendChild(image);
 		}
 	}  else if (data.classname == "story" && data.name == "picture_prompt") {
-		if (document.getElementById("action image").firstChild) {
-			document.getElementById("action image").firstChild.setAttribute("title", data.value);
-		}
+		if (data.value) document.getElementById("action image").setAttribute("tooltip", data.value);
 	//special case for welcome text since we want to allow HTML
 	} else if (data.classname == 'model' && data.name == 'welcome') {
 		document.getElementById('welcome_text').innerHTML = data.value;