Updated tests for control flow: call instruction stores call element
 instead of callee expression