Add unit test to utils