Fix a couple of files that lacked the appropriate header.