Switching to bandit testing framework