Fixed more bugs on the bandit plugin