Add: small enhancement on the raft test