Fix compile errors due to missing header includes.