Fixed a reference to DataSet