The Deputy First Minister told the Senedd he is optimistic that revenue generated by UK-led reforms to water pollution regulations will stay in Wales.