diff --git a/.github/workflows/test.yaml b/.github/workflows/test.yaml
index 88d5e9232..30e95a5e6 100644
--- a/.github/workflows/test.yaml
+++ b/.github/workflows/test.yaml
@@ -18,6 +18,11 @@ defaults:
 jobs:
   test:
     runs-on: ubuntu-latest
+    strategy:
+      fail-fast: false
+      matrix:
+        python-version: ["3.9", "3.10", "3.11"]
+
     steps:
       - uses: actions/checkout@v2
 
@@ -26,6 +31,8 @@ jobs:
         with:
           environment-file: ci/environment.yml
           environment-name: dask-match
+          extra-specs: |
+            python=${{ matrix.python-version }}
 
       - name: Install Dask-Match
         run: python -m pip install -e . --no-deps
diff --git a/dask_expr/collection.py b/dask_expr/collection.py
index 177da760c..e1ab97b27 100644
--- a/dask_expr/collection.py
+++ b/dask_expr/collection.py
@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import functools
 
 import numpy as np
diff --git a/dask_expr/expr.py b/dask_expr/expr.py
index 8485f8dbb..b8617c021 100644
--- a/dask_expr/expr.py
+++ b/dask_expr/expr.py
@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import functools
 import numbers
 import operator
@@ -415,7 +417,7 @@ def __dask_graph__(self):
     def __dask_keys__(self):
         return [(self._name, i) for i in range(self.npartitions)]
 
-    def substitute(self, substitutions: dict) -> "Expr":
+    def substitute(self, substitutions: dict) -> Expr:
         """Substitute specific `Expr` instances within `self`
 
         Parameters