Corrected compilation issue due to missing #include