diff --git a/src/moments_eq_rhs.F90 b/src/moments_eq_rhs.F90
index 8e25ed8fd15eb7f0d2e23a227b805257f623cc25..bdbfc06e201db7fa1a7d7d5a2a030d1a368007bb 100644
--- a/src/moments_eq_rhs.F90
+++ b/src/moments_eq_rhs.F90
@@ -127,6 +127,8 @@ SUBROUTINE moments_eq_rhs
         kzloope : DO ikz = ikzs,ikze
           kr     = krarray(ikr)   ! Poloidal wavevector
           kz     = kzarray(ikz)   ! Toroidal wavevector
+          IF (Nkz .EQ. 1) kz = krarray(ikr) ! If 1D simulation we put kr as kz
+
           kperp2 = kr**2 + kz**2  ! perpendicular wavevector
           be_2   = kperp2 * sigmae2_taue_o2 ! Kernel argument
 
@@ -338,6 +340,8 @@ SUBROUTINE moments_eq_rhs
         kzloopi : DO ikz = ikzs,ikze
           kr     = krarray(ikr)   ! Poloidal wavevector
           kz     = kzarray(ikz)   ! Toroidal wavevector
+          IF (Nkz .EQ. 1) kz = krarray(ikr) ! If 1D simulation we put kr as kz
+          
           kperp2 = kr**2 + kz**2  ! perpendicular wavevector
           bi_2   = kperp2 * sigmai2_taui_o2 ! Kernel argument