Fix noise scaling bug