diff --git a/src/unilab/__init__.py b/src/unilab/__init__.py
index d19618207..2866101eb 100644
--- a/src/unilab/__init__.py
+++ b/src/unilab/__init__.py
@@ -1,7 +1,5 @@
 """UniLab package initialization."""
 
-import os
-
 from etils import epath
 
 __version__ = "0.0.0"
diff --git a/src/unilab/algos/mlx/common/rollout_storage.py b/src/unilab/algos/mlx/common/rollout_storage.py
index b62bf4ab7..b2b876b58 100644
--- a/src/unilab/algos/mlx/common/rollout_storage.py
+++ b/src/unilab/algos/mlx/common/rollout_storage.py
@@ -2,7 +2,7 @@
 
 from __future__ import annotations
 
-from dataclasses import dataclass, field
+from dataclasses import dataclass
 from typing import Dict, Generator, Union
 
 import mlx.core as mx