Fix missing #include