diff --git a/SegmentedControl/Protocols.swift b/SegmentedControl/Protocols.swift
index f11e535..b921332 100644
--- a/SegmentedControl/Protocols.swift
+++ b/SegmentedControl/Protocols.swift
@@ -8,17 +8,7 @@
 
 import UIKit
 
-public protocol SegmentedControlDelegate: class {
-    func segmentedControl(_ segmentedControl: SegmentedControl, didSelectIndex selectedIndex: Int)
-    func segmentedControl(_ segmentedControl: SegmentedControl, didLongPressIndex longPressIndex: Int)
-}
-
-public extension SegmentedControlDelegate {
-    func segmentedControl(_ segmentedControl: SegmentedControl, didSelectIndex selectedIndex: Int) {
-
-    }
-
-    func segmentedControl(_ segmentedControl: SegmentedControl, didLongPressIndex longPressIndex: Int) {
-
-    }
+@objc public protocol SegmentedControlDelegate: NSObjectProtocol {
+    @objc optional func segmentedControl(_ segmentedControl: SegmentedControl, didSelectIndex selectedIndex: Int)
+    @objc optional func segmentedControl(_ segmentedControl: SegmentedControl, didLongPressIndex longPressIndex: Int)
 }
diff --git a/SegmentedControl/SegmentedControl.swift b/SegmentedControl/SegmentedControl.swift
index 8c57128..260c239 100644
--- a/SegmentedControl/SegmentedControl.swift
+++ b/SegmentedControl/SegmentedControl.swift
@@ -262,7 +262,7 @@ open class SegmentedControl: UIControl {
         }
 
         if 0..<segmentsCount() ~= touchIndex {
-            delegate?.segmentedControl(self, didSelectIndex: touchIndex)
+            delegate?.segmentedControl?(self, didSelectIndex: touchIndex)
             if touchIndex != selectedIndex {
                 setSelected(at: touchIndex, animated: isAnimationEnabled)
             }
@@ -419,7 +419,7 @@ extension SegmentedControl: UIGestureRecognizerDelegate {
             return
         }
         if 0..<segmentsCount() ~= longPressIndex {
-            delegate?.segmentedControl(self, didLongPressIndex: longPressIndex)
+            delegate?.segmentedControl?(self, didLongPressIndex: longPressIndex)
         }
     }
 }