2 files changed, 6 insertions, 2 deletions
diff --git a/clang/test/Modules/Inputs/Modified/B.h b/clang/test/Modules/Inputs/Modified/B.h
index d1c8bb5e8e1..f50dd80ca4e 100644
--- a/clang/test/Modules/Inputs/Modified/B.h
+++ b/clang/test/Modules/Inputs/Modified/B.h
@@ -1,2 +1,3 @@
-#include "A.h"
+@import A;
+
 int getB();
diff --git a/clang/test/Modules/Inputs/Modified/module.map b/clang/test/Modules/Inputs/Modified/module.map
index d9aed01430c..50aaebc5c75 100644
--- a/clang/test/Modules/Inputs/Modified/module.map
+++ b/clang/test/Modules/Inputs/Modified/module.map
@@ -1,2 +1,5 @@
 module A { header "A.h" }
-module B { header "B.h" }
+module B { 
+  header "B.h" 
+  export *
+}