Fixed the broken build problem