Use new riak_test helper API