diff --git a/IntelFrameworkPkg/Library/UefiLibFramework/UefiDriverModel.c b/IntelFrameworkPkg/Library/UefiLibFramework/UefiDriverModel.c
index 2153aad337..f6919a6a78 100644
--- a/IntelFrameworkPkg/Library/UefiLibFramework/UefiDriverModel.c
+++ b/IntelFrameworkPkg/Library/UefiLibFramework/UefiDriverModel.c
@@ -13,10 +13,7 @@
 
 **/ 
 
-//
-// Include common header file for this module.
-//
-#include "UefiLibInternal.h"
+#include "UefiLibFramework.h"
 
 /**
   Intialize a driver by installing the Driver Binding Protocol onto the driver's