Add coverage for driver