--- a/tests/testlib/ext-sidedata.py	Sat Dec 05 23:11:49 2020 +0100
+++ b/tests/testlib/ext-sidedata.py	Tue Dec 01 09:13:08 2020 +0100
@@ -15,9 +15,10 @@
     node,
     requirements,
     revlog,
-    upgrade,
 )
 
+from mercurial.upgrade_utils import engine as upgrade_engine
+
 from mercurial.revlogutils import sidedata
 
 
@@ -79,5 +80,5 @@
     extensions.wrapfunction(revlog.revlog, 'addrevision', wrapaddrevision)
     extensions.wrapfunction(revlog.revlog, 'revision', wraprevision)
     extensions.wrapfunction(
-        upgrade, 'getsidedatacompanion', wrapgetsidedatacompanion
+        upgrade_engine, 'getsidedatacompanion', wrapgetsidedatacompanion
     )