Fix noise timesteps vs dimensions.