diff --git a/apps/client/src/widgets/type_widgets/Book.tsx b/apps/client/src/widgets/type_widgets/Book.tsx
index b1ed94d68..6b796f6b6 100644
--- a/apps/client/src/widgets/type_widgets/Book.tsx
+++ b/apps/client/src/widgets/type_widgets/Book.tsx
@@ -16,7 +16,7 @@ export default function Book({ note }: TypeWidgetProps) {
         setShouldDisplayNoChildrenWarning(!note.hasChildren() && VIEW_TYPES.includes(viewType ?? ""));
     }
 
-    useEffect(refresh, [ note ]);
+    useEffect(refresh, [ note, viewType ]);
     useTriliumEvent("entitiesReloaded", ({ loadResults }) => {
         if (loadResults.getBranchRows().some(branchRow => branchRow.parentNoteId === note.noteId)) {
             refresh();