fix bug for some driver