small bug fix
Merge pull request #25 from ErikFang/master

small bug fix