Change internal import to flask_babel