Fix the broken code