Fix data utils import