minor change to data prep