             return settingsLabels.count
         case Section.wordList.rawValue:
             let wordCount = self.mutableCustomCategory.getWordCount()
-
-            // Delete category button will show only for existing custom categories
-            return self.existingCustomCategory ? wordCount + 2 : wordCount + 1
+            return wordCount + 1
+        case Section.deleteCategory.rawValue:
+            // Delete category button will only show for existing categories
+            return self.existingCustomCategory ? 1 : 0
         default:
             return 0
         }

    enum Section: Int {
        case settings = 0
        case wordList = 1
    }

    enum Setting: Int {

                    break
                }
            default:
                break
            }
        case Section.deleteCategory.rawValue:
            if self.existingCustomCategory {
                self.onDeleteCategory()
            }
        default:
            break
