diff --git a/src/memray/_metadata.py b/src/memray/_metadata.py
index 5d5913b498..ab47ea1926 100644
--- a/src/memray/_metadata.py
+++ b/src/memray/_metadata.py
@@ -1,7 +1,12 @@
+from __future__ import annotations
+
+import typing
+
 from dataclasses import dataclass
 from datetime import datetime
 
-from ._memray import FileFormat
+if typing.TYPE_CHECKING:
+    from ._memray import FileFormat
 
 
 @dataclass