diff --git a/src/pre_process/m_start_up.fpp b/src/pre_process/m_start_up.fpp
index 3b45547c29..1f349ab139 100644
--- a/src/pre_process/m_start_up.fpp
+++ b/src/pre_process/m_start_up.fpp
@@ -767,7 +767,7 @@ contains
 
         real(wp), intent(inout) :: start, finish
 
-        integer :: j, k
+        integer :: j, k, l
 
         ! Setting up the grid and the initial condition. If the grid is read in from
         ! preexisting grid data files, it is checked for consistency. If the grid is
@@ -787,10 +787,16 @@ contains
 
         ! hard-coded psi
         if (hyper_cleaning) then
-            do j = 0, m
+            do l = 0, p
                 do k = 0, n
-                    q_cons_vf(psi_idx)%sf(j, k, 0) = 1d-2*exp(-(x_cc(j)**2 + y_cc(k)**2)/(2.0*0.05**2))
-                    q_prim_vf(psi_idx)%sf(j, k, 0) = q_cons_vf(psi_idx)%sf(j, k, 0)
+                    do j = 0, m
+                        if (p > 0) then
+                            q_cons_vf(psi_idx)%sf(j, k, l) = 1.0e-2_wp*exp(-(x_cc(j)**2 + y_cc(k)**2 + z_cc(l)**2)/(2.0_wp*0.05_wp**2))
+                        else
+                            q_cons_vf(psi_idx)%sf(j, k, l) = 1.0e-2_wp*exp(-(x_cc(j)**2 + y_cc(k)**2)/(2.0_wp*0.05_wp**2))
+                        end if
+                        q_prim_vf(psi_idx)%sf(j, k, l) = q_cons_vf(psi_idx)%sf(j, k, l)
+                    end do
                 end do
             end do
         end if