diff --git a/privacy/bolton/losses_test.py b/privacy/bolton/losses_test.py
index 3996ae9..6c60c35 100644
--- a/privacy/bolton/losses_test.py
+++ b/privacy/bolton/losses_test.py
@@ -17,8 +17,8 @@ from __future__ import absolute_import
 from __future__ import division
 from __future__ import print_function
 
-from contextlib import contextmanager  # pylint: disable=importing-member
-from io import StringIO  # pylint: disable=importing-member
+from contextlib import contextmanager  # pylint: disable=g-importing-member
+from io import StringIO  # pylint: disable=g-importing-member
 import sys
 from absl.testing import parameterized
 import tensorflow as tf