fix reg all in one class error