Improve until() helper