1 files changed, 4 insertions, 1 deletions
diff --git a/Tweak.xm b/Tweak.xm
index f2748d6..189b39b 100644
--- a/Tweak.xm
+++ b/Tweak.xm
@@ -91,7 +91,10 @@ void SafeModeButtonClicked(int button) {
 
 void SafeModeButtonClicked(id self, int button) {
     SafeModeButtonClicked(button);
-    [self deactivateForButton];
+    if ([self respondsToSelector:@selector(deactivateForButton)])
+        [self deactivateForButton];
+    else
+        [self dismiss];
 }
 
 void SafeModeAlertItem$alertSheet$buttonClicked$(id self, SEL sel, id sheet, int button) {