Ah, you've found (and fixed) another bug! Could you extend your test to verify that we're doing this correctly?

Some code style changes:
- Could you place the comment on the preceding line--we try to stay under 80 characters in the source files.
- Could you insert a space between the `if` keyword and the open parenthesis (`(`) character?