Fixed integration test issue