Tiny modification
Merge pull request #210 from pmampe/cleanup

Slight modification