1 files changed, 2 insertions, 2 deletions
diff --git a/clang-tools-extra/modularize/ModularizeUtilities.cpp b/clang-tools-extra/modularize/ModularizeUtilities.cpp
index 1bbae3c84ab..62b6e911fbf 100644
--- a/clang-tools-extra/modularize/ModularizeUtilities.cpp
+++ b/clang-tools-extra/modularize/ModularizeUtilities.cpp
@@ -278,14 +278,14 @@ bool ModularizeUtilities::collectModuleHeaders(const Module &Mod) {
       MI != MIEnd; ++MI)
     collectModuleHeaders(**MI);
 
-  if (const FileEntry *UmbrellaHeader = Mod.getUmbrellaHeader()) {
+  if (const FileEntry *UmbrellaHeader = Mod.getUmbrellaHeader().Entry) {
     std::string HeaderPath = getCanonicalPath(UmbrellaHeader->getName());
     // Collect umbrella header.
     HeaderFileNames.push_back(HeaderPath);
 
     // FUTURE: When needed, umbrella header header collection goes here.
   }
-  else if (const DirectoryEntry *UmbrellaDir = Mod.getUmbrellaDir()) {
+  else if (const DirectoryEntry *UmbrellaDir = Mod.getUmbrellaDir().Entry) {
     // If there normal headers, assume these are umbrellas and skip collection.
     if (Mod.Headers->size() == 0) {
       // Collect headers in umbrella directory.