added enforce_Dirichlet_BC() after flux correction