diff --git a/seqio/loggers_test.py b/seqio/loggers_test.py
index 2973eaf9..5e42ced6 100644
--- a/seqio/loggers_test.py
+++ b/seqio/loggers_test.py
@@ -13,10 +13,12 @@
 # limitations under the License.
 
 """Tests for seqio.loggers."""
-# pylint:disable=g-bare-generic,g-long-lambda
 
+# pylint:disable=g-bare-generic,g-long-lambda
 
+import collections
 import dataclasses
+import itertools
 import json
 import os
 from typing import Optional