--- a/hgext/lfs/__init__.py	Thu Dec 07 23:44:06 2017 -0500
+++ b/hgext/lfs/__init__.py	Fri Dec 08 00:18:30 2017 -0500
@@ -140,6 +140,9 @@
 
     wrapfunction(scmutil, 'wrapconvertsink', wrapper.convertsink)
 
+    wrapfunction(upgrade, '_finishdatamigration',
+                 wrapper.upgradefinishdatamigration)
+
     wrapfunction(upgrade, 'preservedrequirements',
                  wrapper.upgraderequirements)