adding more test coverage around Driver