Clean up gradient setup animation logic [swift]

cbpowell · Jul 9, 2016 · 3ee1e76 · 3ee1e76
1 parent f201564
commit 3ee1e76
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/Sources/Swift/MarqueeLabel.swift b/Sources/Swift/MarqueeLabel.swift
@@ -1161,8 +1161,8 @@ public class MarqueeLabel: UILabel {
     }
 
     override public func animationDidStop(anim: CAAnimation, finished flag: Bool) {
-        if anim is GradientSetupAnimation {
-            if let setupAnim = maskLayer?.animationForKey("setupFade") as? CABasicAnimation, finalColors = setupAnim.toValue as? [CGColorRef] {
+        if let setupAnim = anim as? GradientSetupAnimation {
+            if let finalColors = setupAnim.toValue as? [CGColorRef] {
                 maskLayer?.colors = finalColors
             }
             // Remove regardless, since we set removeOnCompletion = false