small updates to dataset feature test