Remove !! from loadCategories() function

Co-Authored-By: Irfan Ömür <[email protected]>
neeldoshii · Jun 19, 2024 · 4cf519e · 4cf519e
1 parent 2ac2694
commit 4cf519e
Showing 1 changed file with 3 additions and 7 deletions.
diff --git a/WordPress/src/main/java/org/wordpress/android/ui/posts/AddCategoryFragment.kt b/WordPress/src/main/java/org/wordpress/android/ui/posts/AddCategoryFragment.kt
@@ -23,8 +23,8 @@ class AddCategoryFragment : AppCompatDialogFragment() {
     private var site: SiteModel? = null
     private var binding: AddCategoryBinding? = null
 
-    @set:Inject
-    var taxonomyStore: TaxonomyStore? = null
+    @Inject
+    lateinit var taxonomyStore: TaxonomyStore
 
     override fun onCreateDialog(savedInstanceState: Bundle?): Dialog {
         (requireActivity().application as WordPress).component().inject(this)
@@ -93,11 +93,7 @@ class AddCategoryFragment : AppCompatDialogFragment() {
     }
 
     private fun loadCategories() {
-        val rootCategory = CategoryNode.createCategoryTreeFromList(
-            taxonomyStore!!.getCategoriesForSite(
-                site!!
-            )
-        )
+        val rootCategory = site?.let { CategoryNode.createCategoryTreeFromList(taxonomyStore.getCategoriesForSite(it)) }
         val categoryLevels = CategoryNode.getSortedListOfCategoriesFromRoot(rootCategory)
         categoryLevels.add(0, CategoryNode(0, 0, getString(R.string.top_level_category_name)))
         if (categoryLevels.size > 0) {