Simplified JITFunction context logic