diff --git a/index.html b/index.html
index 43a39aa..a60c812 100644
--- a/index.html
+++ b/index.html
@@ -381,7 +381,7 @@
src="js/ui/tool/generic.js?v=3e678e0"
type="text/javascript">
-
+
diff --git a/js/ui/tool/dream.js b/js/ui/tool/dream.js
index 6caf2e1..6f3b880 100644
--- a/js/ui/tool/dream.js
+++ b/js/ui/tool/dream.js
@@ -522,6 +522,7 @@ const _generate = async (endpoint, request, bb, options = {}) => {
images.splice(at, 1);
seeds.splice(at, 1);
if (at > images.length - 1) prevImg();
+ if (images.length - 1 === 0) discardImg();
imageindextxt.textContent = `${at}/${images.length - 1}`;
var seed = seeds[at];
seedbtn.title = "Use seed " + seed;