Changing mock to double