TAG: patch for KMS Bandit