diff --git a/src/plumpy/base/state_machine.py b/src/plumpy/base/state_machine.py
index 853ca668..3b9e1c5a 100644
--- a/src/plumpy/base/state_machine.py
+++ b/src/plumpy/base/state_machine.py
@@ -1,5 +1,6 @@
 # -*- coding: utf-8 -*-
 """The state machine for processes"""
+from __future__ import annotations
 
 import enum
 import functools
diff --git a/src/plumpy/process_comms.py b/src/plumpy/process_comms.py
index 39b70d4f..0e2e1e37 100644
--- a/src/plumpy/process_comms.py
+++ b/src/plumpy/process_comms.py
@@ -1,5 +1,6 @@
 # -*- coding: utf-8 -*-
 """Module for process level communication functions and classes"""
+from __future__ import annotations
 
 import asyncio
 import logging