diff --git a/test/sumfact/poisson/poisson_dg_tensor.ufl b/test/sumfact/poisson/poisson_dg_tensor.ufl
index 2734383f71d3afddc39c6c72f41ba5545def3b54..50af4173dd5f1a6a5c744d31f7cbe0f74befe3c5 100644
--- a/test/sumfact/poisson/poisson_dg_tensor.ufl
+++ b/test/sumfact/poisson/poisson_dg_tensor.ufl
@@ -25,15 +25,15 @@ gamma_int = (alpha * degree * (degree + dim - 1)) / h_int
 # SIPG: -1.0, IIPG: 0.0, NIPG: 1.0
 theta = 1.0
 
-r = inner(grad(u), grad(v))*dx \
-  - f*v*dx \
-  - inner(n, avg(grad(u)))*jump(v)*dS \
+r = inner(A*grad(u), grad(v))*dx \
+  - (c*u-f)*v*dx \
+  - inner(n, A*avg(grad(u)))*jump(v)*dS \
   + gamma_int*jump(u)*jump(v)*dS \
-  + theta*jump(u)*inner(avg(grad(v)), n)*dS \
-  - inner(n, grad(u))*v*ds \
+  + theta*jump(u)*inner(A*avg(grad(v)), n)*dS \
+  - inner(n, A*grad(u))*v*ds \
   + gamma_ext*u*v*ds \
-  + theta*u*inner(grad(v), n)*ds \
+  + theta*u*inner(A*grad(v), n)*ds \
   - gamma_ext*g*v*ds \
-  - theta*g*inner(grad(v), n)*ds
+  - theta*g*inner(A*grad(v), n)*ds
 
 exact_solution = g