Fixing build errors for DataTaskIO