bug fix
Merge pull request #416 from LI-HE/master

bug fix