Fixed missing #include