Fix broken compilation