diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
index 4432251c17..5827bd617c 100644
--- a/.pre-commit-config.yaml
+++ b/.pre-commit-config.yaml
@@ -14,7 +14,7 @@ repos:
         exclude: ^requirements-dev\.txt$
     -   id: trailing-whitespace
 - repo: https://github.com/pre-commit/mirrors-mypy
-  rev: v0.971
+  rev: v0.981
   hooks:
     - id: mypy
       name: Run static type checks
diff --git a/pymc/blocking.py b/pymc/blocking.py
index 744b1a3a78..bd58cc38a3 100644
--- a/pymc/blocking.py
+++ b/pymc/blocking.py
@@ -20,7 +20,7 @@
 import collections
 
 from functools import partial
-from typing import Callable, Dict, Optional, TypeVar
+from typing import Callable, Dict, Generic, Optional, TypeVar
 
 import numpy as np
 
@@ -35,7 +35,7 @@
 RaveledVars = collections.namedtuple("RaveledVars", "data, point_map_info")
 
 
-class Compose:
+class Compose(Generic[T]):
     """
     Compose two functions in a pickleable way
     """