1 files changed, 5 insertions, 1 deletions
diff --git a/packages/SystemUI/src/com/android/systemui/usb/UsbPreferenceActivity.java b/packages/SystemUI/src/com/android/systemui/usb/UsbPreferenceActivity.java
index 187a486..7b07f79 100644
--- a/packages/SystemUI/src/com/android/systemui/usb/UsbPreferenceActivity.java
+++ b/packages/SystemUI/src/com/android/systemui/usb/UsbPreferenceActivity.java
@@ -39,6 +39,7 @@ public class UsbPreferenceActivity extends Activity implements View.OnClickListe
     private String mCurrentFunction;
     private String[] mFunctions;
     private String mInstallerImagePath;
+    private AlertDialog mDialog;
     private Button mMtpPtpButton;
     private Button mInstallerCdButton;
     private boolean mPtpActive;
@@ -71,7 +72,7 @@ public class UsbPreferenceActivity extends Activity implements View.OnClickListe
             mInstallerCdButton.setVisibility(View.GONE);
         }
 
-        dialogBuilder.show();
+        mDialog = dialogBuilder.show();
     }
 
     public void onClick(View v) {
@@ -88,6 +89,9 @@ public class UsbPreferenceActivity extends Activity implements View.OnClickListe
             mUsbManager.setMassStorageBackingFile(mInstallerImagePath);
         }
 
+        if (mDialog != null) {
+            mDialog.dismiss();
+        }
         finish();
     }
 }