bug fix
Merge pull request #164 from xiangli-cmu/bug_fix

bug fix