Fix the unit tests by adding missing parameter
Merge pull request #9 from frankdu/fix_unit_test

Fix the unit tests by adding missing parameter