Fix tests
Merge pull request #3 from btbinhtran/fix-tests

Fix tests