Fixed include directive for the trainer