Fix re-connection to Riak after disconnect