fixing integration test bug