Fixing bandit submodule