Fix test failure in ShardTest