Put this into a `if ... end` block; hard to follow like this. Thanks.