Add util tests