diff --git a/configure.ac b/configure.ac
index ae14ae8d3b..c7da7fbe53 100644
--- a/configure.ac
+++ b/configure.ac
@@ -203,7 +203,7 @@ AC_ARG_WITH([tensorflow],
 AM_CONDITIONAL([TENSORFLOW], false)
 TENSORFLOW_LIBS=
 AS_IF([test "x$with_tensorflow" != xno],
-  [AC_CHECK_FILE([/usr/include/tensorflow/core/framework/graph.pb.h],
+  [AC_CHECK_HEADERS(["tensorflow/core/framework/graph.pb.h"],
     [AC_SUBST([TENSORFLOW_LIBS], ["-lprotobuf -ltensorflow_cc"])
       AM_CONDITIONAL([TENSORFLOW], true)
     ],