Fixing small nits identified by the reviewer.