1 files changed, 20 insertions, 17 deletions
diff --git a/Source/FieldSolver/SpectralSolver/SpectralAlgorithms/CMakeLists.txt b/Source/FieldSolver/SpectralSolver/SpectralAlgorithms/CMakeLists.txt
index 912ed47c4..9d414a274 100644
--- a/Source/FieldSolver/SpectralSolver/SpectralAlgorithms/CMakeLists.txt
+++ b/Source/FieldSolver/SpectralSolver/SpectralAlgorithms/CMakeLists.txt
@@ -1,19 +1,22 @@
-target_sources(WarpX
-  PRIVATE
-    PsatdAlgorithmFirstOrder.cpp
-    PsatdAlgorithmJConstantInTime.cpp
-    PsatdAlgorithmJLinearInTime.cpp
-    PsatdAlgorithmPml.cpp
-    SpectralBaseAlgorithm.cpp
-    PsatdAlgorithmComoving.cpp
-)
-
-if(WarpX_DIMS STREQUAL RZ)
-    target_sources(WarpX
+foreach(D IN LISTS WarpX_DIMS)
+    warpx_set_suffix_dims(SD ${D})
+    target_sources(WarpX_${SD}
       PRIVATE
-        SpectralBaseAlgorithmRZ.cpp
-        PsatdAlgorithmRZ.cpp
-        PsatdAlgorithmGalileanRZ.cpp
-        PsatdAlgorithmPmlRZ.cpp
+        PsatdAlgorithmFirstOrder.cpp
+        PsatdAlgorithmJConstantInTime.cpp
+        PsatdAlgorithmJLinearInTime.cpp
+        PsatdAlgorithmPml.cpp
+        SpectralBaseAlgorithm.cpp
+        PsatdAlgorithmComoving.cpp
     )
-endif()
+
+    if(D STREQUAL "RZ")
+        target_sources(WarpX_${SD}
+          PRIVATE
+            SpectralBaseAlgorithmRZ.cpp
+            PsatdAlgorithmRZ.cpp
+            PsatdAlgorithmGalileanRZ.cpp
+            PsatdAlgorithmPmlRZ.cpp
+        )
+    endif()
+endforeach()