The restore-process no longer happens inside onPartitionsAssigned()