Fix restart. Not shuffling images anymore