Update the tests for utils