Correctly clean up with direct driver

Same.