updating mock setup behavior