diff --git a/tests/fixtures/data.py b/tests/fixtures/data.py
index 0b8fe57491..16dd01347d 100644
--- a/tests/fixtures/data.py
+++ b/tests/fixtures/data.py
@@ -2,7 +2,7 @@
 # SPDX-License-Identifier: Apache-2.0
 
 from pathlib import Path
-from unittest.mock import MagicMock
+from unittest.mock import MagicMock, patch
 
 from composer.utils import dist
 from omegaconf import DictConfig