Added back riak_test behaviour.