diff --git a/include/llvm/Target/TargetSchedInfo.h b/include/llvm/Target/TargetSchedInfo.h
index a6a03b52490..1c6008f9078 100644
--- a/include/llvm/Target/TargetSchedInfo.h
+++ b/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;
 
 };