Fixed #include bug