Add missing load statements (#4508) This is required for compatibility with bazel @ HEAD. Done with: `buildifier --lint=fix -r .`