Adding a missing #include