fix tests to adapt to multiple values
Merge pull request #11 from philnguyen/fix-tests

fix tests to adapt to multiple values