--- a/tests/test-annotate.py	Fri Mar 09 21:59:07 2018 -0500
+++ b/tests/test-annotate.py	Wed Feb 28 15:09:05 2018 -0500
@@ -6,7 +6,7 @@
 from mercurial import (
     mdiff,
 )
-from mercurial.context import (
+from mercurial.dagop import (
     annotateline,
     _annotatepair,
 )