Implement auto restore