Update of di mock .