Add Mock DI. Is this really necessary?