Remove inclusion of dataset.h