diff --git a/toolbox/__init__.py b/toolbox/__init__.py
index 22eda81..c763cb9 100644
--- a/toolbox/__init__.py
+++ b/toolbox/__init__.py
@@ -37,6 +37,8 @@ recognized_datasets = [
     "VCTK-Corpus/wav48",
     "aidatatang_200zh/corpus/dev",
     "aidatatang_200zh/corpus/test",
+    "aishell3/test/wav",
+    "magicdata/train",
 ]
 
 #Maximum of generated wavs to keep on memory