Fix visible additional view issue

grigaci · Jan 19, 2016 · b48214c · b48214c
1 parent b63d888
commit b48214c
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/BIObjCHelpers/Datasource/TableView/BIDatasourceFeedTableView.m b/BIObjCHelpers/Datasource/TableView/BIDatasourceFeedTableView.m
@@ -11,6 +11,7 @@
 #import "BIBatchResponse.h"
 #import "BITableViewCell.h"
 #import "_BITableView+Internal.h"
+#import "BITableAdditionalViewBase.h"
 
 @interface BIDatasourceFeedTableView ()
 
@@ -217,6 +218,9 @@ - (void)updateTableAdditionalViewsForBatchResponse:(nonnull BIBatchResponse *)ba
         } else {
             [self.tableView removeVisibleAdditionalView];
         }
+        if (self.tableView.visibleAdditionalView.type == BITableAdditionalTypeLoadingContentView) {
+            [self.tableView removeVisibleAdditionalView];
+        }
     }
 }