diff --git a/tests/test_early_stopping.py b/tests/test_early_stopping.py
index b342494..9fb5f99 100644
--- a/tests/test_early_stopping.py
+++ b/tests/test_early_stopping.py
@@ -1,6 +1,5 @@
 import torch
 
-from torch.utils.data import Dataset
 from khaosz.core import *
 from khaosz.trainer import *
 from khaosz.trainer.data_util import *