diff --git a/tests/test_main.py b/tests/test_main.py
index 35fc2cbb..ed741042 100644
--- a/tests/test_main.py
+++ b/tests/test_main.py
@@ -27,6 +27,7 @@ import PyPDF2 as pypdf
 from ocrmypdf.exceptions import ExitCode
 from ocrmypdf import leptonica
 from ocrmypdf.exec import ghostscript, tesseract, qpdf
+from ocrmypdf.pdfa import file_claims_pdfa
 import logging
 from math import isclose