diff --git a/train_async.py b/train_async.py
index f972c0794bb..9ee86e12553 100644
--- a/train_async.py
+++ b/train_async.py
@@ -53,7 +53,7 @@ def train(args):
 
         if (rollout_id + 1) % args.update_weights_interval == 0:
             # sync generate before update weights to prevent update weight in the middle of generation
-            rollout_data_curr_ref = ray.get(rollout_data_next_future)
+            rollout_data_curr_ref = ray.get(x) if (x := rollout_data_next_future) is not None else None
             rollout_data_next_future = None
             actor_model.update_weights()