Make transitionMapFunction a call_by_name parameter instead of a function.