diff --git a/tensorflow_probability/python/distributions/distribution_properties_test.py b/tensorflow_probability/python/distributions/distribution_properties_test.py
index 25d53b2310..84289baa35 100644
--- a/tensorflow_probability/python/distributions/distribution_properties_test.py
+++ b/tensorflow_probability/python/distributions/distribution_properties_test.py
@@ -341,8 +341,8 @@ def testCanConstructAndSampleDistribution(self, data):
     x = self.evaluate(new_dist.sample(seed=test_util.test_seed()))
     self.assertEqual(sample_shape, x.shape)
 
-    # Valid parameters should give finite samples.
-    self.assertAllFinite(x)
+    # Valid parameters should give non-nan samples.
+    self.assertAllFalse(np.isnan(x))
 
 
 def _all_shapes(thing):