Pull out util module
Fix test