I think this return should be outside the `if`.