diff --git a/BIObjCHelpers/Datasource/TableView/BIDatasourceFeedTableView.m b/BIObjCHelpers/Datasource/TableView/BIDatasourceFeedTableView.m index 99a4927..3ac3034 100644 --- 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]; + } } }