Fixed the broken build