diff --git a/CMakeLists.txt b/CMakeLists.txt
index a1edf2d7a68b1e65e32ad85635729495413f09aa..19d27e519981d145ef623bc2ece05f242ce9083d 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -195,7 +195,7 @@ else()
 endif()
 
 if (OPENMP_FOUND)
-	if (NOT CMAKE_CUDA_TOOLKIT_INCLUDE_DIRECTORIES STREQUAL "")
+	if (NOT OpenMP_CXX_INCLUDE_DIRS STREQUAL "")
 		file(WRITE openmp_flags "${OpenMP_CXX_FLAGS} -I${OpenMP_CXX_INCLUDE_DIRS}")
 	else()
 		file(WRITE openmp_flags "${OpenMP_CXX_FLAGS}")