Fixed compilation (include was missing)