diff --git a/torchsnapshot/io_preparers/sharded_tensor.py b/torchsnapshot/io_preparers/sharded_tensor.py
index b238af0..37411b8 100644
--- a/torchsnapshot/io_preparers/sharded_tensor.py
+++ b/torchsnapshot/io_preparers/sharded_tensor.py
@@ -4,6 +4,7 @@
 #
 # This source code is licensed under the BSD-style license found in the
 # LICENSE file in the root directory of this source tree.
+from __future__ import annotations
 
 import asyncio
 import copy