diff --git a/tests/test_block_exact_match.py b/tests/test_block_exact_match.py
index e59a6dd..fad84ae 100644
--- a/tests/test_block_exact_match.py
+++ b/tests/test_block_exact_match.py
@@ -8,6 +8,7 @@ from hivemind import P2PHandlerError
 from test_utils import *
 
 import src
+from src import DistributedBloomConfig
 from src.bloom.from_pretrained import load_pretrained_block
 from src.client.remote_sequential import RemoteTransformerBlock
 from src.data_structures import UID_DELIMITER