Fix the broken build