diff --git a/tests/helpers.py b/tests/helpers.py
index 2ecb04523..d17868485 100644
--- a/tests/helpers.py
+++ b/tests/helpers.py
@@ -1,3 +1,4 @@
+import contextlib
 import os
 
 from compose.config.config import ConfigDetails