Fix Dataset pickle on Python 2