[SHARDING] Fix tests