Fix the capture.
Small cleanup.