diff --git a/shared/python/trial_util.py b/shared/python/trial_util.py
index 83efdc4..332e364 100644
--- a/shared/python/trial_util.py
+++ b/shared/python/trial_util.py
@@ -17,7 +17,7 @@ def set_seed(seed):
     # need to have their seeds set
     np.random.seed(seed)
     mx.random.seed(seed)
-    tf.set_random_seed(seed)
+    tf.random.set_seed(seed)
     pt.manual_seed(seed)
     random.seed(seed)