diff --git a/invenio_previewer/__init__.py b/invenio_previewer/__init__.py
index d91067bc75e0b77ac1983c44b67b8fbc2f3f2256..8dd99ffe28f04870ea42f2711b9864c6ac4ebbf2 100644
--- a/invenio_previewer/__init__.py
+++ b/invenio_previewer/__init__.py
@@ -74,7 +74,7 @@ Extensions
 Now that we have configured the Flask application, let's initialize all
 dependent Invenio extensions:
 
->>> from flask_babelex import Babel
+>>> from invenio_i18n import Babel
 >>> from invenio_assets import InvenioAssets
 >>> from invenio_db import InvenioDB, db
 >>> from invenio_records import InvenioRecords
diff --git a/tests/conftest.py b/tests/conftest.py
index 07f422a921ddf25c138d92b561f050aaacc9f509..c812cbc9718521ef7141b75d395bc9d8057fd39e 100644
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -19,7 +19,6 @@ from zipfile import ZipFile
 
 import pytest
 from flask import Flask
-from flask_babelex import Babel
 from flask_webpackext import current_webpack
 from invenio_assets import InvenioAssets
 from invenio_config import InvenioConfigDefault
@@ -28,6 +27,7 @@ from invenio_db import db as db_
 from invenio_files_rest import InvenioFilesREST
 from invenio_files_rest.models import Location, ObjectVersion
 from invenio_formatter import InvenioFormatter
+from invenio_i18n import Babel
 from invenio_pidstore.providers.recordid import RecordIdProvider
 from invenio_records import InvenioRecords
 from invenio_records_files.api import Record