diff --git a/CMakeLists.txt b/CMakeLists.txt
index cb09275..7323399 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -67,6 +67,9 @@ add_custom_target(dsl_headers ALL DEPENDS ${DSL_HEADERS})
 if (DOUBLE_PRECISION)
     add_definitions(-DAC_DOUBLE_PRECISION=1)
 endif ()
+if (MPI_ENABLED)
+    add_definitions(-DAC_MPI_ENABLED=1)
+endif ()
 
 add_definitions(-DAC_DEFAULT_CONFIG="${CMAKE_SOURCE_DIR}/config/astaroth.conf")
 
diff --git a/src/core/CMakeLists.txt b/src/core/CMakeLists.txt
index 6e18185..152f811 100644
--- a/src/core/CMakeLists.txt
+++ b/src/core/CMakeLists.txt
@@ -6,7 +6,6 @@ target_link_libraries(astaroth_core astaroth_utils astaroth_kernels cudart)
 if (MPI_ENABLED)
     find_package(MPI)
     target_link_libraries(astaroth_core MPI::MPI_CXX)
-    target_compile_definitions(astaroth_core PRIVATE -DAC_MPI_ENABLED=1)
 endif()
 
 if (MULTIGPU_ENABLED)