fixed minor driver issue