Reparing integration test