Add test dep on mock