Fix rescale problem