Adding support for custom drawing.

1 files changed, 6 insertions, 1 deletions

diff --git a/MASShortcutView.m b/MASShortcutView.m
index 386584f..ba28e47 100644
--- a/MASShortcutView.m
+++ b/MASShortcutView.m
@@ -33,11 +33,16 @@
 
 #pragma mark -
 
++ (Class)shortcutCellClass
+{
+    return [NSButtonCell class];
+}
+
 - (id)initWithFrame:(CGRect)frameRect
 {
     self = [super initWithFrame:frameRect];
     if (self) {
-        _shortcutCell = [[NSButtonCell alloc] init];
+        _shortcutCell = [[[self.class shortcutCellClass] alloc] init];
         _shortcutCell.buttonType = NSPushOnPushOffButton;
         _shortcutCell.font = [[NSFontManager sharedFontManager] convertFont:_shortcutCell.font toSize:BUTTON_FONT_SIZE];
         _enabled = YES;