Update ZedAI break detection to the new API