postgres rollback to functions

I'm now using DBeaver and it have very handy featue: When in "manual commit" mode if you run function from console and it executes another function and so on, when you rollback initial function all the functions that were executed from initial are being rolled back. Is there similar feature in DataGrip ? I've tried to use manual commit mode, but initial function is being rolled back but all other functions are not.

Maybe i'm doing something wrong?