Fix mock