small fix to the data generating script