Add a couple missing headers.