diff --git a/setup.py b/setup.py
index 5110251..4c170e5 100644
--- a/setup.py
+++ b/setup.py
@@ -43,7 +43,7 @@
         'immutabledict',
         'librosa',
         'mir_eval',
-        'note_seq',
+        'note-seq @ git+https://github.com/magenta/note-seq#egg=note_seq',
         'numpy',
         'pretty_midi',
         'scikit-learn',