Updated the test case
https://github.com/zhuchenwang/cluster2/commit/98f4437c552810df090cbdb3fd012abecfd18449