Small logic change
Merge pull request #13 from GantMan/master

super small code change