Fixed problem with try-expect blocks

Please verify. Works for me.