diff --git a/src/mraa.i b/src/mraa.i
index 4810a51..fd77c1c 100644
--- a/src/mraa.i
+++ b/src/mraa.i
@@ -6,7 +6,7 @@
 
 #ifdef DOXYGEN
   //and doesn't seem to work correctly in SWIG's preprocessor
-  #ifdef SWIGPYTHON
+  #ifdef SWIGPYTHON2
         %include common_hpp_doc.i
         %include gpio_class_doc.i
         %include i2c_class_doc.i
diff --git a/src/python/python2/CMakeLists.txt b/src/python/python2/CMakeLists.txt
index 6f1b14d..79d8daa 100644
--- a/src/python/python2/CMakeLists.txt
+++ b/src/python/python2/CMakeLists.txt
@@ -22,7 +22,7 @@ if (PYTHON2_LIBRARY)
 
     set_target_properties (${SWIG_MODULE_python2-mraa_REAL_NAME} PROPERTIES
       OUTPUT_NAME _mraa
-      COMPILE_FLAGS "${CMAKE_C_FLAGS} -DSWIGPYTHON=${SWIG_FOUND}"
+      COMPILE_FLAGS "${CMAKE_C_FLAGS} -DSWIGPYTHON=${SWIG_FOUND} -DSWIGPYTHON2=${SWIG_FOUND}"
     )
 
     install (FILES ${CMAKE_CURRENT_BINARY_DIR}/_mraa${CMAKE_SHARED_LIBRARY_SUFFIX}
diff --git a/src/python/python3/CMakeLists.txt b/src/python/python3/CMakeLists.txt
index 8d688ba..068605f 100644
--- a/src/python/python3/CMakeLists.txt
+++ b/src/python/python3/CMakeLists.txt
@@ -15,7 +15,7 @@ if (PYTHON3_LIBRARY)
 
     set_target_properties (${SWIG_MODULE_python3-mraa_REAL_NAME} PROPERTIES
       OUTPUT_NAME _mraa
-      COMPILE_FLAGS "${CMAKE_C_FLAGS} -DSWIGPYTHON=${SWIG_FOUND}"
+      COMPILE_FLAGS "${CMAKE_C_FLAGS} -DSWIGPYTHON=${SWIG_FOUND} -DDSWIGPYTHON3=${SWIG_FOUND}"
     )
 
     install (FILES ${CMAKE_CURRENT_BINARY_DIR}/_mraa${CMAKE_SHARED_LIBRARY_SUFFIX}