Make the dataset env script a little more friendly.