Fix incorrect shape when adding noise.