fixed bug in driver