Add shard allocation status check