diff --git a/lib/widgets/download_tree_view.dart b/lib/widgets/download_tree_view.dart
index f5a641e6..c36c3683 100644
--- a/lib/widgets/download_tree_view.dart
+++ b/lib/widgets/download_tree_view.dart
@@ -513,8 +513,8 @@ class _DownloadTreeItemState extends State<_DownloadTreeItem> {
   @override
   void didUpdateWidget(_DownloadTreeItem oldWidget) {
     super.didUpdateWidget(oldWidget);
-    // Reinitialize focus nodes if action count might have changed
-    if (oldWidget.node.status != widget.node.status) {
+    // Reinitialize focus nodes if action count changed
+    if (_getActionCount() != _buttonFocusNodes.length) {
       _disposeButtonFocusNodes();
       _initButtonFocusNodes();
     }