diff --git a/src/com/android/settings/enterprise/ActionDisabledByAdminDialogHelper.java b/src/com/android/settings/enterprise/ActionDisabledByAdminDialogHelper.java
index e30f890..ccc7436 100644
--- a/src/com/android/settings/enterprise/ActionDisabledByAdminDialogHelper.java
+++ b/src/com/android/settings/enterprise/ActionDisabledByAdminDialogHelper.java
@@ -80,7 +80,8 @@
             EnforcedAdmin enforcedAdmin) {
         AlertDialog.Builder builder = new AlertDialog.Builder(mActivity)
                 .setPositiveButton(R.string.okay,
-                        mActionDisabledByAdminController.getPositiveButtonListener())
+                        mActionDisabledByAdminController
+                                .getPositiveButtonListener(mActivity, enforcedAdmin))
                 .setView(mDialogView);
         prepareDialogBuilder(builder, restriction, enforcedAdmin);
         return builder;