diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
index b59784dbf..446ba0fe7 100644
--- a/.pre-commit-config.yaml
+++ b/.pre-commit-config.yaml
@@ -4,6 +4,8 @@ repos:
     hooks:
       - id: black
         args: [--line-length=80]
+        additional_dependencies: ['click==8.0.1']
+        exclude: icefall\/__init__\.py
 
   - repo: https://github.com/PyCQA/flake8
     rev: 3.9.2
diff --git a/icefall/__init__.py b/icefall/__init__.py
index f466d6a62..ec77e89b5 100644
--- a/icefall/__init__.py
+++ b/icefall/__init__.py
@@ -1,3 +1,13 @@
+# isort:skip_file
+
+from . import (
+    checkpoint,
+    decode,
+    dist,
+    env,
+    utils
+)
+
 from .checkpoint import (
     average_checkpoints,
     find_checkpoints,
diff --git a/pyproject.toml b/pyproject.toml
index ec5623f90..b4f8c3377 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -10,4 +10,5 @@ exclude = '''
   | \.github
 )/
   | make_kn_lm.py
+  | icefall\/__init__\.py
 '''