fix issue 178
Merge pull request #179 from stevenzwu/issue-178

fix issue 178