Fix compile error.

kirbyt · Mar 6, 2016 · 4c6ff5f · 4c6ff5f
1 parent 441e557
commit 4c6ff5f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/WPSKit/UIKit/UIView+WPSKit.swift b/WPSKit/UIKit/UIView+WPSKit.swift
@@ -26,9 +26,9 @@ extension UIView {
    
    - returns A `UIImage` representation of the view.
    */
-  public func imageSnapshot -> UIImage {
+  public func imageSnapshot() -> UIImage {
     UIGraphicsBeginImageContextWithOptions(self.bounds.size, true, 0.0)
-    self.layer.renderInContext(UIGraphicsGetCurrentContext())
+    self.layer.renderInContext(UIGraphicsGetCurrentContext()!)
     let image = UIGraphicsGetImageFromCurrentImageContext()
     UIGraphicsEndImageContext()