Updated integration tests helper