Added mock method