fixed bug
Merge pull request #2 from w5mith/feature_bug_fix

fixed bug