driver test: fix import