diff --git a/code/ysolve.f90 b/code/ysolve.f90
index 42553ff..27c5b74 100644
--- a/code/ysolve.f90
+++ b/code/ysolve.f90
@@ -108,8 +108,7 @@
       nd=1 
       CALL d2fnonp(nx,hx,ux(1,:),d2ux(1,:),nd,1) 
 
-      dm = diff
-      where (y1(:,1,1)<(1-cstar)) dm = diff * rvis
+      dm = diff * (1.0 + (rvis - 1.0) * (1.0 - y1(:,1,1)))
 
       DO i=1,nx
        yr=ux(1,i)
@@ -389,8 +388,7 @@
         CALL dfnonp(nx,hx,ux(1:3,:),dux(1:3,:),3,1)
         CALL d2fnonp(nx,hx,ux(3:3,:),d2ux(1,:),1,1)
 
-      dm = diff
-      where (r1(:,1,1)<(1.0-cstar)) dm = diff * rvis
+      dm = diff * (1.0 + (rvis - 1.0) * (1.0 - r1(:,1,1)))
 
         DO i=1,nx
          wrate=pre*ux(3,i)*exp(-ac/(1.+bc*(1.-ux(3,i))))   !wrate