diff --git a/ultravox/data/datasets.py b/ultravox/data/datasets.py
index d6707d87..4ed3413b 100644
--- a/ultravox/data/datasets.py
+++ b/ultravox/data/datasets.py
@@ -1,7 +1,5 @@
 import abc
 import dataclasses
-import enum
-import io
 import logging
 import os
 import tempfile
diff --git a/ultravox/data/datasets_test.py b/ultravox/data/datasets_test.py
index ace96dd0..98e4c6f6 100644
--- a/ultravox/data/datasets_test.py
+++ b/ultravox/data/datasets_test.py
@@ -1,4 +1,4 @@
-from typing import Optional, Union
+from typing import Optional
 
 import datasets as hf_datasets
 import numpy as np