diff --git a/code/ysolve.f90 b/code/ysolve.f90
index 5a14813..0f3d650 100644
--- a/code/ysolve.f90
+++ b/code/ysolve.f90
@@ -589,8 +589,8 @@
 
         ! Boundary conditions
         IF (i.eq.nx) THEN
-            f(nx,1) = -wrate1 - u(nx)*dux(1,nx)
-            f(nx,2) = wrate1 - u(nx)*dux(2,nx)
+            f(nx,1) = -wrate - u(nx)*dux(1,nx)
+            f(nx,2) = wrate - u(nx)*dux(2,nx)
         ENDIF
       ENDDO