1 files changed, 4 insertions, 3 deletions
diff --git a/Source/FieldSolver/SpectralSolver/SpectralAlgorithms/PsatdAlgorithmRZ.H b/Source/FieldSolver/SpectralSolver/SpectralAlgorithms/PsatdAlgorithmRZ.H
index 9d6ed9cfa..4987e88d7 100644
--- a/Source/FieldSolver/SpectralSolver/SpectralAlgorithms/PsatdAlgorithmRZ.H
+++ b/Source/FieldSolver/SpectralSolver/SpectralAlgorithms/PsatdAlgorithmRZ.H
@@ -18,14 +18,12 @@ class PsatdAlgorithmRZ : public SpectralBaseAlgorithmRZ
     public:
         PsatdAlgorithmRZ(SpectralKSpaceRZ const & spectral_kspace,
                          amrex::DistributionMapping const & dm,
+                         const SpectralFieldIndex& spectral_index,
                          int const n_rz_azimuthal_modes, int const norder_z,
                          bool const nodal, amrex::Real const dt_step,
                          bool const update_with_rho);
         // Redefine functions from base class
         virtual void pushSpectralFields(SpectralFieldDataRZ & f) override final;
-        virtual int getRequiredNumberOfFields() const override final {
-            return SpectralFieldIndex::n_fields;
-        }
 
         void InitializeSpectralCoefficients(SpectralFieldDataRZ const & f);
 
@@ -62,6 +60,9 @@ class PsatdAlgorithmRZ : public SpectralBaseAlgorithmRZ
                                     std::array<std::unique_ptr<amrex::MultiFab>,3>& current) override final;
 
     private:
+
+        SpectralFieldIndex m_spectral_index;
+
         bool coefficients_initialized;
         // Note that dt is saved to use in InitializeSpectralCoefficients
         amrex::Real m_dt;