Updated coverage of dataset tests