Fix integration bug