Fix the unit tests by adding missing parameter