MacGui: fix closing the add catagery window when presented as modal.

1 files changed, 18 insertions, 2 deletions

diff --git a/macosx/HBAddCategoryController.m b/macosx/HBAddCategoryController.m
index dd85ea4d7..598f70460 100644
--- a/macosx/HBAddCategoryController.m
+++ b/macosx/HBAddCategoryController.m
@@ -52,7 +52,15 @@
 
 - (IBAction)dismiss:(id)sender
 {
-    [self.window.sheetParent endSheet:self.window returnCode:NSModalResponseCancel];
+    if (self.window.isSheet)
+    {
+        [self.window.sheetParent endSheet:self.window returnCode:NSModalResponseCancel];
+    }
+    else
+    {
+        [NSApp stopModalWithCode:NSModalResponseCancel];
+        [self.window orderOut:self];
+    }
 }
 
 - (IBAction)create:(id)sender
@@ -60,7 +68,15 @@
     self.category = [[HBPreset alloc] initWithCategoryName:self.name.stringValue builtIn:NO];
     [self.manager addPreset:self.category];
 
-    [self.window.sheetParent endSheet:self.window returnCode:NSModalResponseOK];
+    if (self.window.isSheet)
+    {
+        [self.window.sheetParent endSheet:self.window returnCode:NSModalResponseOK];
+    }
+    else
+    {
+        [NSApp stopModalWithCode:NSModalResponseOK];
+        [self.window orderOut:self];
+    }
 }