diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml
index 8e50f5a6..92f06cdf 100644
--- a/.github/workflows/main.yml
+++ b/.github/workflows/main.yml
@@ -92,10 +92,19 @@ jobs:
         uses: actions/setup-python@v4
       - name: black
         uses: psf/black@stable
+        with:
+          version: 23.12.1
+          jupyter: true
       - name: isort
         uses: isort/isort-action@master
         with:
             configuration: "--profile black"
+            isort-version: 5.12.0
+      - name: autoflake
+        uses: creyD/autoflake_action@master
+        with:
+            options: --remove-all-unused-imports --in-place
+            # Pegging version s.t. https://github.com/creyD/autoflake_action/issues/1
       - name: Install and configure Poetry
         uses: snok/install-poetry@v1
         with:
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
index 4136db4c..b931949f 100644
--- a/.pre-commit-config.yaml
+++ b/.pre-commit-config.yaml
@@ -1,11 +1,11 @@
 repos:
 - repo: https://github.com/psf/black
-  rev: 23.12.1
+  rev: 23.12.1  # NOTE: If updating, also update the version in .github/workflows/*
   hooks:
     - id: black-jupyter
       types_or: ["python","jupyter"] 
 - repo: https://github.com/pycqa/isort
-  rev: 5.12.0
+  rev: 5.12.0  # NOTE: If updating, also update the version in .github/workflows/*
   hooks:
     - id: isort
 - repo: https://github.com/PyCQA/autoflake