fix minor bug
Merge pull request #393 from lunika/minorFix

fix minor bug