Addressed a comment from the review