add wait decorator