fixed bad mock method