Small improvement on a nested statement

Here's my very small contribution to the project. I saw an comment about nested stuff, and modified it in a way I think is a bit more idiomatic Rust code.