Return mock on tests