Fixed broken compilation