recover datasets.py changes