diff --git a/libs/DTBExtnLib/DTBExtnLib_node.c b/libs/DTBExtnLib/DTBExtnLib_node.c
index 84e5413..de402e2 100644
--- a/libs/DTBExtnLib/DTBExtnLib_node.c
+++ b/libs/DTBExtnLib/DTBExtnLib_node.c
@@ -16,9 +16,9 @@
 
 #if defined (TARGET_UEFI)
   #include <Library/PrintLib.h>
-  #include <Library/QcomBaseLib.h>
   #include <Library/MemoryAllocationLib.h>
   #include <Library/DebugLib.h>
+  #include <Library/DtFrameworkLibSupport.h>
 #elif defined (TARGET_XBL)
   #include <stdint.h>
   #include <stdio.h>