--- a/hgext/lfs/wrapper.py	Sat Dec 05 23:11:49 2020 +0100
+++ b/hgext/lfs/wrapper.py	Tue Dec 01 09:13:08 2020 +0100
@@ -34,6 +34,8 @@
     wireprotov1server,
 )
 
+from mercurial.upgrade_utils import engine as upgrade_engine
+
 from mercurial.interfaces import repository
 
 from mercurial.utils import (
@@ -520,7 +522,7 @@
     remoteblob.writebatch(pointers, repo.svfs.lfslocalblobstore)
 
 
-@eh.wrapfunction(upgrade, b'_finishdatamigration')
+@eh.wrapfunction(upgrade_engine, b'_finishdatamigration')
 def upgradefinishdatamigration(orig, ui, srcrepo, dstrepo, requirements):
     orig(ui, srcrepo, dstrepo, requirements)