Fixed not awaiting to save unit of work.