diff --git a/tests/jraphphysics/dataset/test_xdmf_dataset.py b/tests/jraphphysics/dataset/test_xdmf_dataset.py
index c2d3caa..c5b586c 100644
--- a/tests/jraphphysics/dataset/test_xdmf_dataset.py
+++ b/tests/jraphphysics/dataset/test_xdmf_dataset.py
@@ -1,7 +1,4 @@
 import unittest
-import tensorflow as tf
-import numpy as np
-import jax.numpy as jnp
 from torch.utils import data
 
 from jraphphysics.dataset.xdmf_dataset import XDMFDataset