add wait till set function