minor changes to test_20news_dataset.py