diff --git a/src/viscoelastic3d.f90 b/src/viscoelastic3d.f90
index 17d34fd..e26b405 100644
--- a/src/viscoelastic3d.f90
+++ b/src/viscoelastic3d.f90
@@ -123,17 +123,18 @@ END SUBROUTINE viscoelasticdeviatoricstress
 #ifdef USING_CUDA
   SUBROUTINE viscouseigenstress(mu,structure,ductilezones,nz,sig,sx1,sx2,sx3, &
        dx1,dx2,dx3,moment,beta,maxwelltime,gamma)
+    INTEGER, INTENT(IN) :: sx1,sx2,sx3
     TYPE(WEAK_STRUCT), DIMENSION(nz), INTENT(IN) :: ductilezones
     INTEGER, INTENT(IN) :: nz
     REAL*8, INTENT(IN) :: beta      
 #else       
   SUBROUTINE viscouseigenstress(mu,structure,sig,sx1,sx2,sx3, &
        dx1,dx2,dx3,moment,maxwelltime,dgammadot0,gamma)
+    INTEGER, INTENT(IN) :: sx1,sx2,sx3
     REAL*4, DIMENSION(sx1,sx2,sx3), INTENT(IN), OPTIONAL :: dgammadot0
 #endif
     REAL*8, INTENT(IN) :: mu,dx1,dx2,dx3
     TYPE(LAYER_STRUCT), DIMENSION(:), INTENT(IN) :: structure
-    INTEGER, INTENT(IN) :: sx1,sx2,sx3
     TYPE(TENSOR), INTENT(IN), DIMENSION(sx1,sx2,sx3) :: sig
     TYPE(TENSOR), INTENT(OUT), DIMENSION(sx1,sx2,sx3) :: moment
     REAL*8, OPTIONAL, INTENT(INOUT) :: maxwelltime