updated tests for dataset