diff --git a/test/stokes/stokes_dg.ufl b/test/stokes/stokes_dg.ufl
index 5704c65a40bb6c6a450741ccda8b158603598e54..71ff6674ec3f1c09b115c11ecc748a39d0c311c5 100644
--- a/test/stokes/stokes_dg.ufl
+++ b/test/stokes/stokes_dg.ufl
@@ -8,10 +8,20 @@ u, p = TrialFunctions(TH)
 
 n = FacetNormal(cell)('+')
 eps = -1.0
+sigma = 1.0
 
-r = \
-  + eps*inner(avg(grad(v))*n, jump(u))*dS
-#  + inner(grad(u), grad(v))*dx \
-#  - inner(avg(grad(u))*n, jump(v))*dS \
+r = inner(grad(u), grad(v))*dx \
+  - inner(avg(grad(u))*n, jump(v))*dS \
+  + eps * inner(avg(grad(v))*n, jump(u))*dS \
+  - inner(grad(u)*n, v)*ds \
+  + eps * inner(grad(v)*n, u)*ds \
+  + sigma * inner(jump(u), jump(v))*dS \
+  + sigma * inner(u, v)*ds \
+  - p*div(v)*dx \
+  - q*div(u)*dx \
+  + avg(p)*inner(jump(v), n)*dS \
+  + avg(q)*inner(jump(u), n)*dS \
+  + p*inner(v, n)*ds \
+  + q*inner(u, n)*ds
 
 forms = [r]