Fix mock warnings