Do you need a similar change to the if branch above?