Update riak_test.go