Fix to handle edge case.