Switch to use the random.shuffle function