diff --git a/translator/__init__.py b/translator/__init__.py
index 41a2d58..659bdac 100644
--- a/translator/__init__.py
+++ b/translator/__init__.py
@@ -1,6 +1,6 @@
 from translator.translate import Translator
 from translator.language import get_nllb_lang
 
-__version__ = "0.2.5b1"
+__version__ = "0.2.5b2"
 
 LANGS = get_nllb_lang()
diff --git a/translator/main.py b/translator/main.py
index f00cbd2..4a2e7d9 100644
--- a/translator/main.py
+++ b/translator/main.py
@@ -8,7 +8,7 @@
 from datasets import load_dataset, Dataset
 from halo import Halo
 from translator import Translator, utils, __version__
-from translator.languages import get_nllb_lang, get_sys_lang_format
+from translator.language import get_nllb_lang, get_sys_lang_format
 
 locale.setlocale(locale.LC_ALL, '')