Added missing #include so that it builds correctly