Correctly clean up with direct driver

Update the comments on this one.