Add in test for util