fix test which was broken by refactoring