diff --git a/cropharvest/datasets.py b/cropharvest/datasets.py
index ca34dea..f37a70f 100644
--- a/cropharvest/datasets.py
+++ b/cropharvest/datasets.py
@@ -26,7 +26,7 @@
 from cropharvest.engineer import TestInstance
 from cropharvest import countries
 
-from typing import cast, List, Optional, Tuple, Generator, Union
+from typing import cast, List, Optional, Tuple, Generator
 
 
 @dataclass