Fix for data dep error. Maybe