Fix per reivewer's comment.