Switch to bandit testing framework