Fixed bad code in driver