Can you use a random_state here?