Restore task should have its own class