fix up things
Merge pull request #1 from danielzhangyihao/fix-up

fix up things