Fix from review comment.

Silly code, you can be simpler!