diff --git a/llvm/CMakeLists.txt b/llvm/CMakeLists.txt
index 37218cd1093b10..68cd2def61ed73 100644
--- a/llvm/CMakeLists.txt
+++ b/llvm/CMakeLists.txt
@@ -803,8 +803,10 @@ if (TENSORFLOW_C_LIB_PATH)
     endif()
     string(REGEX MATCH "Location: ([^\n]*\n)" TF_PIP_LOC "${TF_PIP_OUT}")
     string(REPLACE "Location: " "" TF_PIP ${TF_PIP_LOC})
-    set(TF_PROTO_HEADERS ${TF_PIP}/tensorflow/include)
+    string(STRIP ${TF_PIP} TF_PIP)
+    set(TF_PROTO_HEADERS "${TF_PIP}/tensorflow/include")
   endif()
+  message(STATUS "Using Tensorflow headers under: ${TF_PROTO_HEADERS}")
   include_directories(${TF_PROTO_HEADERS})
   add_definitions("-DGOOGLE_PROTOBUF_NO_RTTI")
   add_definitions("-D_GLIBCXX_USE_CXX11_ABI=0")