Fix vulnerability issue in driver