Fix migration breakage on the server