diff --git a/java/src/com/android/inputmethod/latin/personalization/PersonalizationDictionarySessionRegister.java b/java/src/com/android/inputmethod/latin/personalization/PersonalizationDictionarySessionRegister.java
index 5b83820197edf161d6c2b543a8d39b2aba19a52b..da59333f5013e98822c461ad0f20931e148a375a 100644
--- a/java/src/com/android/inputmethod/latin/personalization/PersonalizationDictionarySessionRegister.java
+++ b/java/src/com/android/inputmethod/latin/personalization/PersonalizationDictionarySessionRegister.java
@@ -31,5 +31,4 @@ public class PersonalizationDictionarySessionRegister {
 
     public static void onRemoveData(Context context, String type) {
     }
-
 }
diff --git a/java/src/com/android/inputmethod/latin/personalization/PersonalizationDictionaryUpdateSession.java b/java/src/com/android/inputmethod/latin/personalization/PersonalizationDictionaryUpdateSession.java
index 858aa32df5d08d8ecd7e2bcec1a0f97169366b80..433c69c1c55ed158a080fc514bd92ee756b10260 100644
--- a/java/src/com/android/inputmethod/latin/personalization/PersonalizationDictionaryUpdateSession.java
+++ b/java/src/com/android/inputmethod/latin/personalization/PersonalizationDictionaryUpdateSession.java
@@ -16,6 +16,8 @@
 
 package com.android.inputmethod.latin.personalization;
 
+import android.content.Context;
+
 import java.lang.ref.WeakReference;
 import java.util.ArrayList;
 
@@ -48,7 +50,7 @@ public abstract class PersonalizationDictionaryUpdateSession {
 
     public abstract void onDictionaryReady();
 
-    public abstract void onDictionaryClosed();
+    public abstract void onDictionaryClosed(Context context);
 
     public void setPredictionDictionary(String locale, DynamicPredictionDictionaryBase dictionary) {
         mPredictionDictionary = new WeakReference<DynamicPredictionDictionaryBase>(dictionary);
@@ -68,9 +70,9 @@ public abstract class PersonalizationDictionaryUpdateSession {
     }
 
 
-    public void closeSession() {
+    public void closeSession(Context context) {
         unsetPredictionDictionary();
-        onDictionaryClosed();
+        onDictionaryClosed(context);
     }
 
     public void addBigramToPersonalizationDictionary(String word0, String word1, boolean isValid,