modify Mock for first test