Super small fix. Cleanup.
Merge pull request #253 from dovy/master

Single class fix. Want to keep the class clean!