Small improvement to code style