little refactor of integration test.