Fix mock test