Fixed a missing #include.