Improve set within_comment timing