fix up changes
Merge pull request #3 from igfarm/fixup

fix up changes