diff --git a/src/main/frontend/db/restore.cljs b/src/main/frontend/db/restore.cljs
index 57b4767f7f..6fc51b9f5c 100644
--- a/src/main/frontend/db/restore.cljs
+++ b/src/main/frontend/db/restore.cljs
@@ -90,6 +90,9 @@
     (doseq [block data]
       (let [uuid (gobj/get block "uuid")
             eid (get uuid->db-id-map uuid)
+            _ (when (nil? eid)
+                (prn "Error: block without eid ")
+                (js/console.dir block))
             _ (assert eid (str "Can't find eid " eid ", block: " block))
             avs (->> (gobj/get block "datoms")
                      (transit/read t-reader))]
@@ -110,7 +113,8 @@
 
       (p/let [_ (p/delay 150)]          ; More time for UI refresh
         (state/set-state! [repo :restore/unloaded-blocks] nil)
-        (state/set-state! [repo :restore/unloaded-pages] nil)))))
+        (state/set-state! [repo :restore/unloaded-pages] nil)
+        (state/pub-event! [:ui/re-render-root])))))
 
 (defn- uuid-string?
   [s]
diff --git a/src/main/frontend/handler/events.cljs b/src/main/frontend/handler/events.cljs
index 6020eded18..798ffb1476 100644
--- a/src/main/frontend/handler/events.cljs
+++ b/src/main/frontend/handler/events.cljs
@@ -880,6 +880,9 @@
   (p/let [_ (file-handler/alter-file repo path content {:from-disk? true})]
     (ui-handler/re-render-root!)))
 
+(defmethod handle :ui/re-render-root [[_]]
+  (ui-handler/re-render-root!))
+
 (rum/defcs file-id-conflict-item <
   (rum/local false ::resolved?)
   [state repo file data]