Remove the return here and convert the if below into an else if