Fix test for DataflowMappings