error fix in driver, remove mock