Fix test cases
Merge pull request #25 from Rise-Vision/chore/fix-test-cases

Fix test cases