fixing integration testing