add the restore method to RockitModelTrait