fix some bug in heat2d