diff --git a/test/wenet/dataset/test_datapipes.py b/test/wenet/dataset/test_datapipes.py
index 748c749bd..35f4fa5b0 100644
--- a/test/wenet/dataset/test_datapipes.py
+++ b/test/wenet/dataset/test_datapipes.py
@@ -34,4 +34,4 @@ def test_dynamic_batch_datapipe(data_list):
         padding)
 
     for d in dataset:
-        assert d['feats'].size(0) <= max_frames_in_batch
+        assert d['feats'].size(1) <= max_frames_in_batch
diff --git a/test/wenet/dataset/test_dataset.py b/test/wenet/dataset/test_dataset.py
index c5e514920..d9419e5f5 100644
--- a/test/wenet/dataset/test_dataset.py
+++ b/test/wenet/dataset/test_dataset.py
@@ -19,7 +19,6 @@ def test_WenetRawDatasetSource(data_list):
     result = []
     for elem in dataset:
         result.append(elem)
-    print(result)
 
     assert len(result) == len(expected)
     for (i, elem) in enumerate(result):