diff --git a/cmake/tensorrt.cmake b/cmake/tensorrt.cmake
index 8f65a737c43a124c05574d6eb9c3050fdab5299a..fa0e834a1dfd6e60f0ec07945be9a4d84017316f 100644
--- a/cmake/tensorrt.cmake
+++ b/cmake/tensorrt.cmake
@@ -18,7 +18,7 @@ find_library(TENSORRT_LIBRARY NAMES libnvinfer.so libnvinfer.a
 if(TENSORRT_INCLUDE_DIR AND TENSORRT_LIBRARY)
   if(WITH_DSO)
     set(TENSORRT_FOUND ON)
-  endif(WITH DSO)
+  endif(WITH_DSO)
 else()
     set(TENSORRT_FOUND OFF)
 endif()