From 15e537c0f07277e08c3eb4a5a5440d23610664ed Mon Sep 17 00:00:00 2001 From: Joseph Donegan Date: Wed, 7 Oct 2015 11:39:04 +0100 Subject: [PATCH] Fix for https://github.com/iSofTom/STCollapseTableView/issues/9 --- STCollapseTableView/STCollapseTableView.m | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/STCollapseTableView/STCollapseTableView.m b/STCollapseTableView/STCollapseTableView.m index 9a3b893..30d9b6a 100644 --- a/STCollapseTableView/STCollapseTableView.m +++ b/STCollapseTableView/STCollapseTableView.m @@ -91,7 +91,7 @@ - (void)setDataSource:(id )newDataSource - (void)setDelegate:(id)newDelegate { - if (newDelegate != self.collapseDelegate) + if (newDelegate && newDelegate != self.collapseDelegate) { self.collapseDelegate = newDelegate; [super setDelegate:self.collapseDelegate?self:nil];