Fix integration spec failure