Fixed more issues with calling context of computation expressions.