Changed to unit of work property injection