Invariant preservation in machine learned PDE solvers via error correction