Fixing compilation due to wrong headers.