diff --git a/deepwalk/__main__.py b/deepwalk/__main__.py
index bb5ebb69a45656e0847b4f55f2a4a66d02621ef9..1070e074cd6e4125b7186ceb2aedf83026c62acf 100644
--- a/deepwalk/__main__.py
+++ b/deepwalk/__main__.py
@@ -10,10 +10,10 @@ from collections import Counter
 from concurrent.futures import ProcessPoolExecutor
 import logging
 
-import graph
-import walks as serialized_walks
+from . import graph
+from . import walks as serialized_walks
 from gensim.models import Word2Vec
-from skipgram import Skipgram
+from .skipgram import Skipgram
 
 from six import text_type as unicode
 from six import iteritems
diff --git a/deepwalk/walks.py b/deepwalk/walks.py
index 7e90e502fcca62787015f5baedfcf1f57035c8c5..e8f56514e4d17e40691b89c64441b29e2055491d 100644
--- a/deepwalk/walks.py
+++ b/deepwalk/walks.py
@@ -9,7 +9,7 @@ from collections import Counter
 
 from six.moves import zip
 
-import graph
+from . import graph
 
 logger = logging.getLogger("deepwalk")