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 <UITableViewDataSource>)newDataSource
 
 - (void)setDelegate:(id<UITableViewDelegate>)newDelegate
 {
-    if (newDelegate != self.collapseDelegate)
+    if (newDelegate && newDelegate != self.collapseDelegate)
     {
         self.collapseDelegate = newDelegate;
         [super setDelegate:self.collapseDelegate?self:nil];