address review comment 2