diff --git a/js/main.js b/js/main.js
index 67f60279..2ee079d6 100644
--- a/js/main.js
+++ b/js/main.js
@@ -1105,3 +1105,11 @@ document.fonts.addEventListener("loadingdone", (event) => {
   // Initial call to set the margin based on current state
   updateMargin();
 })();
+
+//------------------------------------------------
+// Better scrollbar for Image AI > Gallery
+//------------------------------------------------
+const element = document.querySelector('#image-history-gallery > :nth-child(2)');
+if (element) {
+  element.classList.add('pretty_scrollbar');
+}