Correctly clean up with direct driver

This comment got lost somehow.