diff --git a/requirements.txt b/requirements.txt
index 3374aeb7..dca7decc 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -10,7 +10,7 @@ nvidia-dali-cuda110>=1.34.0
 omegaconf~=2.2.0
 pandas~=1.5.1
 psutil~=5.9.8
-pydftracer==1.0.2
+pydftracer==1.0.8
 pytest
 tensorflow>=2.11.0
 torch>=2.2.0
diff --git a/setup.py b/setup.py
index f8d61560..9b271fdd 100644
--- a/setup.py
+++ b/setup.py
@@ -16,7 +16,7 @@
     "omegaconf>=2.2.0",
     "pandas>=1.5.1",
     "psutil>=5.9.8",
-    "pydftracer==1.0.2",
+    "pydftracer==1.0.8",
 ]
 x86_deps = [
     f"hydra-core>={HYDRA_VERSION}",
diff --git a/tests/dlio_benchmark_test.py b/tests/dlio_benchmark_test.py
index 9de83f4a..d4914fe0 100644
--- a/tests/dlio_benchmark_test.py
+++ b/tests/dlio_benchmark_test.py
@@ -117,7 +117,7 @@ def test_gen_data(fmt, framework) -> None:
         clean()
     finalize()
 
-@pytest.mark.timeout(60, method="thread")
+@pytest.mark.timeout(120, method="thread")
 def test_subset() -> None:
     init()
     clean()
@@ -135,7 +135,7 @@ def test_subset() -> None:
                             '++workload.dataset.num_files_train=8', \
                             '++workload.train.computation_time=0.01'])
         benchmark=run_benchmark(cfg, verify=True)
-    clean()
+        clean()
     finalize()
 
 @pytest.mark.timeout(60, method="thread")