diff --git a/CMakeLists.txt b/CMakeLists.txt
index d8c4aeed..c0c2eb00 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -151,7 +151,7 @@ endif()
 if(WITH_THRIFT)
   # Find Thrift
   find_package(Thrift REQUIRED)
-  add_definitions(-DTHRIFT_ON)
+  add_definitions(-DWITH_THRIFT)
 endif()
 
 if(WITH_PI)
diff --git a/targets/simple_switch_grpc/CMakeLists.txt b/targets/simple_switch_grpc/CMakeLists.txt
index 59daad33..34ce7e40 100644
--- a/targets/simple_switch_grpc/CMakeLists.txt
+++ b/targets/simple_switch_grpc/CMakeLists.txt
@@ -60,7 +60,6 @@ target_link_libraries(simple_switch_grpc PRIVATE
 )
 
 if(WITH_THRIFT)
-  add_definitions(-DWITH_THRIFT)
   target_link_libraries(simple_switch_grpc PRIVATE
          thrift
          simpleswitch_thrift