Small fix to DataController