Fix the scaling bug