diff --git a/llvm/include/llvm/Target/TargetSchedInfo.h b/llvm/include/llvm/Target/TargetSchedInfo.h
index a6a03b52490c..1c6008f90786 100644
--- a/llvm/include/llvm/Target/TargetSchedInfo.h
+++ b/llvm/include/llvm/Target/TargetSchedInfo.h
@@ -280,7 +280,10 @@ protected:
 
 
   friend class ModuloSchedulingPass;
+  friend class ModuloSchedulingSBPass;
   friend class MSSchedule;
+  friend class MSScheduleSB;
+  friend class MSchedGraphSB;
 
 };