Adding repository references to Unit work class