diff --git a/src/ui/party-ui-handler.ts b/src/ui/party-ui-handler.ts
index d4ce186698b..9ea6dab6e3e 100644
--- a/src/ui/party-ui-handler.ts
+++ b/src/ui/party-ui-handler.ts
@@ -1245,7 +1245,9 @@ export default class PartyUiHandler extends MessageUiHandler {
           this.addCommonOptions(pokemon);
         }
         break;
-      // TODO: This still needs to be broken up
+      // TODO: This still needs to be broken up.
+      // It could use a rework differentiating different kind of switches
+      // to treat baton passing separately from switching on faint.
       case PartyUiMode.SWITCH:
       case PartyUiMode.FAINT_SWITCH:
       case PartyUiMode.POST_BATTLE_SWITCH: