Fix broken code formatting