I can pick up naturally generated water sources just fine. But when I pour water to turn a lava lake to obsidian for example, I can't pick the water source back up.

You need to click on the block below the water for it to work. If you just click while there is air behind the water, it doesn't.