Fixed usage of SAMPLE_DATASET_ID in test