fix bug
Merge pull request #120 from qhzhou/staging

fix bug