diff --git a/MacPass/MPContextButton.h b/MacPass/MPContextButton.h
index d118e5b1..15524f12 100644
--- a/MacPass/MPContextButton.h
+++ b/MacPass/MPContextButton.h
@@ -31,7 +31,6 @@ typedef NS_ENUM(NSUInteger, MPContextButtonSegment) {
 };
 
 @property (nonatomic, strong) NSMenu *contextMenu;
-
-- (void)setImage:(NSImage *)image;
+@property (nonatomic, strong) NSImage *image;
 
 @end
diff --git a/MacPass/MPContextButton.m b/MacPass/MPContextButton.m
index 6c95fa77..36f6320b 100644
--- a/MacPass/MPContextButton.m
+++ b/MacPass/MPContextButton.m
@@ -91,6 +91,10 @@
   [self setImage:image forSegment:MPContextButtonSegmentButton];
 }
 
+- (NSImage *)image {
+  return [self imageForSegment:MPContextButtonSegmentButton];
+}
+
 - (void)showContextMenu:(id)sender {
   NSPoint point = self.frame.origin;
   point.x = [self.cell widthForSegment:MPContextButtonSegmentButton];