diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
index 2c1374c..145edb8 100644
--- a/.github/workflows/build.yml
+++ b/.github/workflows/build.yml
@@ -35,7 +35,7 @@ jobs:
       run: |
         python -m pip install poetry
         poetry lock --no-update
-        poetry install -E dev -E rnn
+        poetry install -E dev -E rnn -E visuals
     - name: Verify that we can build the package
       run: poetry build
     - name: Test with pytest