Fixing error edge case.