diff --git a/include/llvm/Target/TargetSchedInfo.h b/include/llvm/Target/TargetSchedInfo.h
index 8a0530d3ed7..fd88667ee40 100644
--- a/include/llvm/Target/TargetSchedInfo.h
+++ b/include/llvm/Target/TargetSchedInfo.h
@@ -302,8 +302,8 @@ protected:
 			   conflictLists;   // indexed by [opcode]
 
 
-
-  friend class ModuloScheduling;
+  friend class ModuloSchedGraph;
+  friend class ModuloSchedulingPass;
   
 };