NOBUG: Added support for after_restore() methods on restore. Will help some blocks.