BUG: fix data.py regression