Added mock in the generator