Stick this assignment with the other ones after the `if`?