BugFix: remember complete dataset instead of label (can be undefined)