diff --git a/utils/dataset_manager.py b/utils/dataset_manager.py
index 0c7e114..1aa0b87 100644
--- a/utils/dataset_manager.py
+++ b/utils/dataset_manager.py
@@ -6,6 +6,8 @@ Dataset management functions for organizing and processing datasets.
 import os
 import shutil
 import json
+import tempfile
+import zipfile
 from typing import Dict, List, Optional
 from pathlib import Path