When reloading dynamic grid, reload restore page first.

Change-Id: I05c1d1faf61ecd79abf15d4ab84a689b7e0cb2ac

1 files changed, 6 insertions, 2 deletions

diff --git a/src/com/android/launcher3/Launcher.java b/src/com/android/launcher3/Launcher.java
index 34f1cd07d..0f80f8cbb 100644
--- a/src/com/android/launcher3/Launcher.java
+++ b/src/com/android/launcher3/Launcher.java
@@ -5541,6 +5541,10 @@ public class Launcher extends Activity
     }
 
     public void updateDynamicGrid() {
+        updateDynamicGrid(mWorkspace.getRestorePage());
+    }
+
+    public void updateDynamicGrid(int page) {
         mSearchDropTargetBar.setupQSB(Launcher.this);
 
         initializeDynamicGrid();
@@ -5548,7 +5552,7 @@ public class Launcher extends Activity
         mGrid.layout(Launcher.this);
 
         // Synchronized reload
-        mModel.startLoader(true, mWorkspace.getCurrentPage());
+        mModel.startLoader(true, page);
         mWorkspace.updateCustomContentVisibility();
 
     }
@@ -5559,7 +5563,7 @@ public class Launcher extends Activity
 
     public boolean updateGridIfNeeded() {
         if (mDynamicGridUpdateRequired) {
-            updateDynamicGrid();
+            updateDynamicGrid(mWorkspace.getCurrentPage());
             mDynamicGridUpdateRequired = false;
             return true;
         }