diff --git a/CMakeLists.txt b/CMakeLists.txt
index 562aa81a9e7bceb19fdd49516d13aec166ec1d25..48c592f318443633a238e4e4b61f0e0a9e9f05e0 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -326,7 +326,11 @@ if(NOT XDMF_BUILD_CORE_ONLY)
   endif()
 
   if(XDMF_WRAP_PYTHON)
-    XDMF_SWIG_PYTHON(${XDMF_LIBNAME} XdmfCore XdmfDSM)
+    if (XDMF_BUILD_DSM)
+      XDMF_SWIG_PYTHON(${XDMF_LIBNAME} XdmfCore)
+    else()
+      XDMF_SWIG_PYTHON(${XDMF_LIBNAME} XdmfCore XdmfDSM)
+    endif()
     set(XDMF_PYTHON_INSTALL_DIR ${CMAKE_INSTALL_PREFIX}/lib/python/)
   endif()