fix test refactoring to work with Sauce base class