Fix broken code