Fixed typo in data.py