Update to bandit testing framework