fix bug
Merge pull request #53 from duonglan/fix_bug

fix bug