diff --git a/offline_tasks/deepwalk/__init__.py b/offline_tasks/deepwalk/__init__.py
new file mode 100644
index 0000000..8385722
--- /dev/null
+++ b/offline_tasks/deepwalk/__init__.py
@@ -0,0 +1,8 @@
+"""
+DeepWalk package for graph embedding
+"""
+from .deepwalk import DeepWalk
+from .alias import create_alias_table, alias_sample
+
+__all__ = ['DeepWalk', 'create_alias_table', 'alias_sample']
+
diff --git a/offline_tasks/deepwalk/deepwalk.py b/offline_tasks/deepwalk/deepwalk.py
index 7908620..14e6a51 100644
--- a/offline_tasks/deepwalk/deepwalk.py
+++ b/offline_tasks/deepwalk/deepwalk.py
@@ -3,7 +3,7 @@ import numpy as np
 import networkx as nx
 from joblib import Parallel, delayed
 import itertools
-from alias import create_alias_table, alias_sample
+from .alias import create_alias_table, alias_sample
 from tqdm import tqdm
 import argparse
 import multiprocessing
--
libgit2 0.21.2