Added TODO for class method to return correct break type