diff --git a/examples/Distributions/DiscreteDistributionLabeled.ipynb b/examples/Distributions/DiscreteDistributionLabeled.ipynb
index 3b5bb2049..72cf032b6 100644
--- a/examples/Distributions/DiscreteDistributionLabeled.ipynb
+++ b/examples/Distributions/DiscreteDistributionLabeled.ipynb
@@ -24,6 +24,7 @@
     "from HARK.distribution import (\n",
     "    MeanOneLogNormal,\n",
     "    DiscreteDistributionLabeled,\n",
+    "    calc_expectation,\n",
     "    combine_indep_dstns,\n",
     ")"
    ]
diff --git a/ruff.toml b/ruff.toml
index 924133f20..84d7322cf 100644
--- a/ruff.toml
+++ b/ruff.toml
@@ -1,2 +1,3 @@
 include = ["*.ipynb"]
-ignore = ["E731", "E721", "E402", "F841", "F821", "F405", "F403"]
+# ignore F401 for now: https://github.com/astral-sh/ruff/issues/8354
+ignore = ["E731", "E721", "E402", "F841", "F821", "F405", "F403", "F401"]