1 files changed, 21 insertions, 1 deletions
diff --git a/src/com/android/settings/SetupChooseLockGeneric.java b/src/com/android/settings/SetupChooseLockGeneric.java
index c451ea4..def8593 100644
--- a/src/com/android/settings/SetupChooseLockGeneric.java
+++ b/src/com/android/settings/SetupChooseLockGeneric.java
@@ -18,7 +18,8 @@ package com.android.settings;
 
 import com.android.setupwizard.navigationbar.SetupWizardNavBar;
 
-
+import android.content.Context;
+import android.content.Intent;
 import android.content.res.Resources;
 import android.os.Bundle;
 import android.preference.PreferenceFragment;
@@ -81,5 +82,24 @@ public class SetupChooseLockGeneric extends ChooseLockGeneric
             super.onViewCreated(view, savedInstanceState);
             SetupWizardUtils.setHeaderText(getActivity(), getActivity().getTitle());
         }
+
+        @Override
+        protected Intent getLockPasswordIntent(Context context, int quality, boolean isFallback,
+                int minLength, int maxLength, boolean requirePasswordToDecrypt,
+                boolean confirmCredentials) {
+            final Intent intent = SetupChooseLockPassword.createIntent(context, quality,
+                    isFallback, minLength, maxLength, requirePasswordToDecrypt, confirmCredentials);
+            SetupWizardUtils.copySetupExtras(getActivity().getIntent(), intent);
+            return intent;
+        }
+
+        @Override
+        protected Intent getLockPatternIntent(Context context, boolean isFallback,
+                boolean requirePassword, boolean confirmCredentials) {
+            final Intent intent = SetupChooseLockPattern.createIntent(context, isFallback,
+                    requirePassword, confirmCredentials);
+            SetupWizardUtils.copySetupExtras(getActivity().getIntent(), intent);
+            return intent;
+        }
     }
 }