Small fix cdb_driver.py