compatibility with dataset api change