Can you please elaborate on why we need to handle both cases here?