Add dependency on mock