Fixed Button image

zacharyclaysmith · Mar 12, 2015 · 8d93111 · 8d93111
1 parent 9388259
commit 8d93111
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/AwfulBindingUI/BindableButton.swift b/AwfulBindingUI/BindableButton.swift
@@ -56,8 +56,8 @@ public class BindableButton:UIButton{
         self.hidden = newValue
     }
 
+    //TODO: segment by control state
     private var _imageBinding:BindableValue<UIImage?>?
-
     public var imageBinding:BindableValue<UIImage?>?{
         get{
             return _imageBinding
@@ -73,6 +73,6 @@ public class BindableButton:UIButton{
     }
 
     private func imageBinding_valueChanged(newValue:UIImage?){
-        self.imageView?.image = newValue
+        self.setImage(newValue, forState: UIControlState.Normal)
     }
 }