1 files changed, 8 insertions, 5 deletions
diff --git a/Source/Python/CMakeLists.txt b/Source/Python/CMakeLists.txt
index 00c9e02f1..e10b8486c 100644
--- a/Source/Python/CMakeLists.txt
+++ b/Source/Python/CMakeLists.txt
@@ -1,5 +1,8 @@
-target_sources(WarpX
-  PRIVATE
-    WarpX_py.cpp
-    WarpXWrappers.cpp
-)
+foreach(D IN LISTS WarpX_DIMS)
+    warpx_set_suffix_dims(SD ${D})
+    target_sources(WarpX_${SD}
+      PRIVATE
+        WarpX_py.cpp
+        WarpXWrappers.cpp
+    )
+endforeach()