* feat: regression tests for optional call * chore: address code review * chore: fix assertion * chore: set context rather than return context