diff --git a/modules/ui/tools/save.js b/modules/ui/tools/save.js
index 89fe4ea5b..1efa5f910 100644
--- a/modules/ui/tools/save.js
+++ b/modules/ui/tools/save.js
@@ -104,6 +104,7 @@ export function uiToolSave(context) {
 
 
         context.history()
+            .on('restore.save', updateCount)
             .on('change.save', function(difference) {
                 if (difference) updateCount();
             });