Wrap driver code in closure.

Avoid message spill in the logs.