Fix missing header which breaks Linux build