Merge pull request MortimerGoro#181 from longlongjump/master

Fix button target removing
BenyT · Mar 20, 2016 · bb55e6a · bb55e6a
2 parents 303f446 + 0576a07
commit bb55e6a
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/MGSwipeTableCell/MGSwipeTableCell.m b/MGSwipeTableCell/MGSwipeTableCell.m
@@ -88,7 +88,13 @@ -(instancetype) initWithButtons:(NSArray*) buttonsArray direction:(MGSwipeDirect
         for (UIView * button in _buttons) {
             if ([button isKindOfClass:[UIButton class]]) {
                 UIButton * btn = (UIButton*)button;
-                [btn removeTarget:nil action:@selector(buttonClicked:) forControlEvents:UIControlEventTouchUpInside]; //Remove all targets to avoid problems with reused buttons among many cells
+
+                for (id target in btn.allTargets) {
+                    if ([target isKindOfClass:self.class]) {
+                        [btn removeTarget:target action:@selector(buttonClicked:) forControlEvents:UIControlEventTouchUpInside]; //Remove all targets to avoid problems with reused buttons among many cells
+                    }
+                }
+
                 [btn addTarget:self action:@selector(buttonClicked:) forControlEvents:UIControlEventTouchUpInside];
             }
             if (!differentWidth) {