Small nit - this line could probably go inside the `if` below.