diff --git a/tests/adv_diff_2d.c b/tests/adv_diff_2d.c
index b5a7f53..31e2281 100644
--- a/tests/adv_diff_2d.c
+++ b/tests/adv_diff_2d.c
@@ -85,7 +85,7 @@ int main(int argc,char **argv)
   PetscCall(DMDASetUniformCoordinates(da, 0.0, L_x, 0.0, L_y, 0.0, 0.0));
   PetscCall(KSPSetDM(ksp,(DM)da));
   // We generate the matrix ourselves
-  PetscCall(KSPSetDMActive(ksp, PETSC_FALSE));
+  PetscCall(KSPSetDMActive(ksp, KSP_DMACTIVE_ALL, PETSC_FALSE));
 
   // Create empty matrix and vectors
   PetscCall(DMCreateMatrix(da, &A));
@@ -322,4 +322,4 @@ PetscErrorCode ComputeMat(DM da, Mat A, PetscScalar u, PetscScalar v, PetscScala
   PetscCall(MatAssemblyEnd(A,MAT_FINAL_ASSEMBLY));
 
   return PETSC_SUCCESS;
-}
\ No newline at end of file
+}