fixing integration test