add withClass for FluentTestWaiter