diff --git a/code/m_calculate.f90 b/code/m_calculate.f90
index 6b1dc34..cc8fa43 100644
--- a/code/m_calculate.f90
+++ b/code/m_calculate.f90
@@ -93,11 +93,11 @@ contains
         integer :: i, j ,k
 
 
-        call tp2(zsrc, src, nyp, nzp*nxp)
+        call tp2(zsrc, src, nyp*nzp, nxp)
 
-        do k = 1,nzp
+        do i = 1,nxp
 
-        call dfp(nzp, hzp, zsrc, zdst, nyp, 3)
+        call dfp(nzp, hzp, zsrc(:,:,i), zdst(:,:,i), nyp, 3)
 
         end do
 
@@ -149,11 +149,11 @@ contains
         integer :: i, j ,k
 
 
-        call tp2(zsrc, src, nyp, nzp*nxp)
+        call tp2(zsrc, src, nyp*nzp, nxp)
 
-        do k = 1,nzp
+        do i = 1,nxp
 
-        call d2fp(nzp, hzp, zsrc, zdst, nyp, 3)
+        call d2fp(nzp, hzp, zsrc(:,:,i), zdst(:,:,i), nyp, 3)
 
         end do