Two small modifications.