add support for restore_schema