diff --git a/docs/source/conf.py b/docs/source/conf.py
index 33acabfd7acb..a30ed49e8fd8 100644
--- a/docs/source/conf.py
+++ b/docs/source/conf.py
@@ -15,12 +15,13 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+import glob
 import os
 import re
 import sys
-import glob
 
 import sphinx_book_theme
+from package_info import __version__
 
 # If extensions (or modules to document with autodoc) are in another directory,
 # add these directories to sys.path here. If the directory is relative to the
@@ -30,7 +31,6 @@
 sys.path.insert(0, os.path.abspath("../../nemo"))
 sys.path.insert(0, os.path.abspath("../../nemo_text_processing"))
 
-from package_info import __version__
 
 templates_path = ["_templates"]
 
diff --git a/setup.py b/setup.py
index 89871858ad46..3ada70d21d64 100644
--- a/setup.py
+++ b/setup.py
@@ -18,16 +18,15 @@
 """Setup for pip package."""
 
 import codecs
+import importlib.util
 import os
 import subprocess
 from distutils import cmd as distutils_cmd
 from distutils import log as distutils_log
 from itertools import chain
-import importlib.util
 
 import setuptools
 
-
 spec = importlib.util.spec_from_file_location('package_info', 'nemo/package_info.py')
 package_info = importlib.util.module_from_spec(spec)
 spec.loader.exec_module(package_info)