Addressed code review comment.