Update unit tests for helper