INFRA-39 Changing call to needs_migration? so that it works