Merge pull request #29 from wizeline/issue/28

Fixes Issue #28
WizelineLabs · May 9, 2019 · f0a5112 · f0a5112
2 parents 306bb1d + 3d4020a
commit f0a5112
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/WLEmptyState/Classes/Protocols/WLEmptyStateProtocol.swift b/WLEmptyState/Classes/Protocols/WLEmptyStateProtocol.swift
@@ -53,7 +53,7 @@ extension WLEmptyStateProtocol where Self: UITableView {
 
     var numberOfItems: Int {
         var items = 0
-        let numberOfSections = dataSource?.numberOfSections?(in: self) ?? 0
+        let numberOfSections = dataSource?.numberOfSections?(in: self) ?? 1
         for section in 0..<numberOfSections {
             items += dataSource?.tableView(self, numberOfRowsInSection: section) ?? 0
         }
@@ -66,7 +66,7 @@ extension WLEmptyStateProtocol where Self: UICollectionView {
 
     var numberOfItems: Int {
         var items = 0
-        let numberOfSections = dataSource?.numberOfSections?(in: self) ?? 0
+        let numberOfSections = dataSource?.numberOfSections?(in: self) ?? 1
         for section in 0..<numberOfSections {
             items += dataSource?.collectionView(self, numberOfItemsInSection: section) ?? 0
         }