NIT: Looks like this whole block was never indented properly. Should be another 2 spaces in since it's in an `if` block. Mind fixing it?