Rename all refs of chunks to shards in tests
Fix imports