Add TODO for edge case