diff --git a/setup.py b/setup.py
index c45995b27..80d13f8e2 100644
--- a/setup.py
+++ b/setup.py
@@ -207,6 +207,7 @@ def get_short_description():
     long_description_content_type='text/markdown',
     url='https://github.com/k2-fsa/k2',
     package_dir={
+        '': 'k2/python',
         'k2': 'k2/python/k2',
         'k2.ragged': 'k2/python/k2/ragged',
         'k2.sparse': 'k2/python/k2/sparse',