1 files changed, 4 insertions, 4 deletions
diff --git a/tensorflow/contrib/distributions/python/kernel_tests/bijectors/affine_linear_operator_test.py b/tensorflow/contrib/distributions/python/kernel_tests/bijectors/affine_linear_operator_test.py
index 0738754b21..d4f83567e4 100644
--- a/tensorflow/contrib/distributions/python/kernel_tests/bijectors/affine_linear_operator_test.py
+++ b/tensorflow/contrib/distributions/python/kernel_tests/bijectors/affine_linear_operator_test.py
@@ -21,7 +21,7 @@ from __future__ import print_function
 import numpy as np
 
 from tensorflow.contrib import linalg
-from tensorflow.contrib.distributions.python.ops.bijectors.affine_linear_operator import AffineLinearOperator
+from tensorflow.contrib.distributions.python.ops.bijectors import affine_linear_operator as affine_linear_operator_lib
 from tensorflow.python.platform import test
 
 
@@ -29,7 +29,7 @@ class AffineLinearOperatorTest(test.TestCase):
 
   def testIdentity(self):
     with self.test_session():
-      affine = AffineLinearOperator(
+      affine = affine_linear_operator_lib.AffineLinearOperator(
           validate_args=True)
       x = np.array([[1, 0, -1], [2, 3, 4]], dtype=np.float32)
       y = x
@@ -48,7 +48,7 @@ class AffineLinearOperatorTest(test.TestCase):
       diag = np.array([[1, 2, 3],
                        [2, 5, 6]], dtype=np.float32)
       scale = linalg.LinearOperatorDiag(diag, is_non_singular=True)
-      affine = AffineLinearOperator(
+      affine = affine_linear_operator_lib.AffineLinearOperator(
           shift=shift, scale=scale, validate_args=True)
 
       x = np.array([[1, 0, -1], [2, 3, 4]], dtype=np.float32)
@@ -73,7 +73,7 @@ class AffineLinearOperatorTest(test.TestCase):
                         [4, 3, 2]]],
                       dtype=np.float32)
       scale = linalg.LinearOperatorTriL(tril, is_non_singular=True)
-      affine = AffineLinearOperator(
+      affine = affine_linear_operator_lib.AffineLinearOperator(
           shift=shift, scale=scale, validate_args=True)
 
       x = np.array([[[1, 0, -1],