diff --git a/convlab2/__init__.py b/convlab2/__init__.py
index 0fe7d5bf01a7ac06fff476149427712f8c7c7ee4..a7b4c3c2d972cbb2dab7b091a099e8591b50895c 100755
--- a/convlab2/__init__.py
+++ b/convlab2/__init__.py
@@ -6,7 +6,8 @@ from convlab2.policy import Policy
 from convlab2.nlg import NLG
 from convlab2.dialog_agent import Agent, PipelineAgent
 from convlab2.dialog_agent import Session, BiSession, DealornotSession
-from convlab2.util.unified_datasets_util import load_dataset, load_database
+from convlab2.util.unified_datasets_util import load_dataset, load_database, load_unified_data, \
+    load_nlu_data, load_dst_data, load_policy_data, load_nlg_data, load_e2e_data
 
 from os.path import abspath, dirname