From 1092ddb125f946555f9cf1b54d3caabe5ae6a08a Mon Sep 17 00:00:00 2001
From: Bryan Housel <bryan@mapbox.com>
Date: Fri, 19 Apr 2019 10:56:00 -0400
Subject: [PATCH] Make sure to update the save count when restoring changes too

---
 modules/ui/tools/save.js | 1 +
 1 file changed, 1 insertion(+)

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();
             });