diff --git a/4_evaluation/project/generate_dataset.py b/4_evaluation/project/generate_dataset.py
index cb531d3f..e0dbd6ed 100644
--- a/4_evaluation/project/generate_dataset.py
+++ b/4_evaluation/project/generate_dataset.py
@@ -2,6 +2,7 @@
 import os
 from pydantic import BaseModel, Field
 from datasets import Dataset
+from typing import List
 
 from distilabel.llms import InferenceEndpointsLLM
 from distilabel.pipeline import Pipeline