migrate: Fix calls to functions