Updated code snippets for uow registration