fix bug in driver