Update tests add util method