1 files changed, 4 insertions, 2 deletions
diff --git a/Source/FieldSolver/SpectralSolver/SpectralAlgorithms/GalileanPsatdAlgorithmRZ.H b/Source/FieldSolver/SpectralSolver/SpectralAlgorithms/GalileanPsatdAlgorithmRZ.H
index 3c8d3d60e..18a7f2dcd 100644
--- a/Source/FieldSolver/SpectralSolver/SpectralAlgorithms/GalileanPsatdAlgorithmRZ.H
+++ b/Source/FieldSolver/SpectralSolver/SpectralAlgorithms/GalileanPsatdAlgorithmRZ.H
@@ -21,7 +21,8 @@ class GalileanPsatdAlgorithmRZ : public SpectralBaseAlgorithmRZ
                                   int const n_rz_azimuthal_modes, int const norder_z,
                                   bool const nodal,
                                   const amrex::Array<amrex::Real,3>& v_galilean,
-                                  amrex::Real const dt_step);
+                                  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 {
@@ -63,9 +64,10 @@ class GalileanPsatdAlgorithmRZ : public SpectralBaseAlgorithmRZ
         // Note that dt and v_galilean are saved to use in InitializeSpectralCoefficients
         amrex::Real const m_dt;
         amrex::Array<amrex::Real,3> m_v_galilean;
+        bool m_update_with_rho;
 
         SpectralRealCoefficients C_coef, S_ck_coef;
-        SpectralComplexCoefficients Theta2_coef, X1_coef, X2_coef, X3_coef, X4_coef;
+        SpectralComplexCoefficients Theta2_coef, T_rho_coef, X1_coef, X2_coef, X3_coef, X4_coef;
 
 };