Refactors integration tests