Fix typo in data.py