diff --git a/Example/Objective-C/MZFormSheetPresentationController Objective-C Example/CustomTransition.h b/Example/Objective-C/MZFormSheetPresentationController Objective-C Example/CustomTransition.h
index fb610bf..175cea8 100644
--- a/Example/Objective-C/MZFormSheetPresentationController Objective-C Example/CustomTransition.h	
+++ b/Example/Objective-C/MZFormSheetPresentationController Objective-C Example/CustomTransition.h	
@@ -9,5 +9,5 @@
 #import <Foundation/Foundation.h>
 #import "MZTransition.h"
 
-@interface CustomTransition : NSObject <MZFormSheetPresentationViewControllerTransitionProtocol>
+@interface CustomTransition : MZTransition <MZFormSheetPresentationViewControllerTransitionProtocol>
 @end
diff --git a/MZFormSheetPresentationController/MZFormSheetPresentationViewControllerAnimator.h b/MZFormSheetPresentationController/MZFormSheetPresentationViewControllerAnimator.h
index 6cecf20..9a07250 100644
--- a/MZFormSheetPresentationController/MZFormSheetPresentationViewControllerAnimator.h
+++ b/MZFormSheetPresentationController/MZFormSheetPresentationViewControllerAnimator.h
@@ -33,7 +33,7 @@ extern CGFloat const MZFormSheetPresentationViewControllerAnimatorDefaultTransit
 @property (nonatomic, assign, getter=isPresenting) BOOL presenting;
 @property (nonatomic, assign, getter=isInteractive) BOOL interactive;
 @property (nonatomic, assign) CGFloat transitionDuration;
-@property (nonatomic, strong) MZTransition *transition;
+@property (nonatomic, strong) id<MZFormSheetPresentationViewControllerTransitionProtocol> transition;
 
 + (instancetype)animatorForTransitionStyle:(MZFormSheetPresentationTransitionStyle)transitionStyle;
 @end
diff --git a/MZFormSheetPresentationController/MZTransition.h b/MZFormSheetPresentationController/MZTransition.h
index b0dfb18..2e5b918 100644
--- a/MZFormSheetPresentationController/MZTransition.h
+++ b/MZFormSheetPresentationController/MZTransition.h
@@ -64,9 +64,13 @@ typedef NS_ENUM(NSInteger, MZFormSheetPresentationTransitionStyle) {
 
 @end
 
+#if __IPHONE_OS_VERSION_MAX_ALLOWED < 100000
 @interface MZTransition
-    : NSObject <MZFormSheetPresentationViewControllerTransitionProtocol>
-
+: NSObject <MZFormSheetPresentationViewControllerTransitionProtocol>
+#else
+@interface MZTransition
+: NSObject <MZFormSheetPresentationViewControllerTransitionProtocol, CAAnimationDelegate>
+#endif
 /**
  *  Register custom transition animation style.
  *  You need to setup transitionStyle to MZFormSheetTransitionStyleCustom.