Simplify a bit of code
Merge pull request #27 from bbatsov/minor-change

Simplify a bit of code