fix normalisation 2d model problem