adjust for API changes: use new load_with_labels functions