Fix typo in gen_data.py.