Add class check to isValid

0thernet · Mar 28, 2014 · 65c6521 · 65c6521
1 parent 3544f44
commit 65c6521
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/BZGFormViewController/BZGFormViewController.m b/BZGFormViewController/BZGFormViewController.m
@@ -264,9 +264,11 @@ - (void)formCell:(BZGFormCell *)formCell didChangeValidationState:(BZGValidation
 {
     BOOL isValid = YES;
     for (BZGFormCell *cell in self.formCells) {
-        isValid = isValid &&
-                  (cell.validationState == BZGValidationStateValid ||
-                   cell.validationState == BZGValidationStateWarning);
+        if ([cell isKindOfClass:[BZGFormCell class]]) {
+            isValid = isValid &&
+                      (cell.validationState == BZGValidationStateValid ||
+                       cell.validationState == BZGValidationStateWarning);
+        }
     }
     self.isValid = isValid;
 }