Update test_cluster.go