Correcting my script output to make sure it is returning the correct dataset.