diff --git a/ddsp/losses_test.py b/ddsp/losses_test.py
index 54a8893a..dd4a2302 100644
--- a/ddsp/losses_test.py
+++ b/ddsp/losses_test.py
@@ -33,8 +33,8 @@ def test_output_shape_is_correct(self):
         loudness_weight=1.0,
     )
 
-    input_audio = tf.random.uniform((3, 8000), dtype=tf.float32)
-    target_audio = tf.random.uniform((3, 8000), dtype=tf.float32)
+    input_audio = tf.ones((3, 8000), dtype=tf.float32)
+    target_audio = tf.ones((3, 8000), dtype=tf.float32)
 
     loss = loss_obj(input_audio, target_audio)
 
@@ -47,8 +47,8 @@ class PretrainedCREPEEmbeddingLossTest(tf.test.TestCase):
   def test_output_shape_is_correct(self):
     loss_obj = losses.PretrainedCREPEEmbeddingLoss()
 
-    input_audio = tf.random.uniform((3, 16000), dtype=tf.float32)
-    target_audio = tf.random.uniform((3, 16000), dtype=tf.float32)
+    input_audio = tf.ones((3, 16000), dtype=tf.float32)
+    target_audio = tf.ones((3, 16000), dtype=tf.float32)
 
     loss = loss_obj(input_audio, target_audio)