diff --git a/app/src/util/history.ts b/app/src/util/history.ts
index 3a28fa79e..7d0a3decd 100644
--- a/app/src/util/history.ts
+++ b/app/src/util/history.ts
@@ -355,7 +355,7 @@ export const openHistory = () => {
                         });
                     } else if (dataType === "doc") {
                         fetchPost("/api/history/rollbackDocHistory", {
-                            notebook: (firstPanelElement.querySelectorAll(".b3-select")[1] as HTMLSelectElement).value,
+                            notebook: (firstPanelElement.querySelector('.b3-select[data-type="notebookselect"]') as HTMLSelectElement).value,
                             historyPath: target.parentElement.getAttribute("data-path")
                         });
                     } else if (dataType === "notebook") {