Fix integration spec