Use task_rng.choose to simplify random selection logic