diff --git a/.github/workflows/format.yaml b/.github/workflows/format.yaml
index 13092d5..e1488e9 100644
--- a/.github/workflows/format.yaml
+++ b/.github/workflows/format.yaml
@@ -21,9 +21,6 @@ jobs:
           with:
             python-version: "3.10"
 
-        - name: isort Import Sorting Check
-          uses: isort/isort-action@master
-
         - name: Back Formatting Check
           uses: psf/black@stable
           with:
@@ -36,3 +33,15 @@ jobs:
 
         - name: Lint with Flake8
           uses: py-actions/flake8@v2
+
+        - name: Install Flake8 for Jupyter Notebooks
+          run: |
+            pip install --upgrade pip
+            pip install flake8-nb
+
+        - name: Run Flake8 for Jupyter Notebooks
+          run: |
+            flake8-nb .
+
+        - name: Ensure Clean Notebooks
+          uses: ResearchSoftwareActions/EnsureCleanNotebooksAction@1.1
diff --git a/spectrogram_segmentation.ipynb b/spectrogram_segmentation.ipynb
index 1888ace..5a7d08e 100644
--- a/spectrogram_segmentation.ipynb
+++ b/spectrogram_segmentation.ipynb
@@ -961,6 +961,7 @@
    "outputs": [],
    "source": [
     "# Grab the first NR_LTE signal\n",
+    "PATH_TO_MATLAB5G_TRAINING_DATA = os.getcwd()\n",
     "spec_path = os.path.join(PATH_TO_MATLAB5G_TRAINING_DATA, \"LTE_NR\", \"LTE_NR_frame_0.png\")\n",
     "spectrogram = read_image(spec_path)  # Image has both NR and LTE signal\n",
     "spectrogram = spectrogram.to(device)\n",