diff --git a/CodeApp/Views/TableView.swift b/CodeApp/Views/TableView.swift
index 22531128..461b272d 100644
--- a/CodeApp/Views/TableView.swift
+++ b/CodeApp/Views/TableView.swift
@@ -169,6 +169,7 @@ where
             || context.coordinator.lastTheme?.id != themeManager.currentTheme?.id
         {
             uiViewController.setData(data: root)
+            context.coordinator.lastTheme = themeManager.currentTheme
         }
         if let header, header != uiViewController.headerText {
             uiViewController.setHeaderText(text: header)