Fix to visitor after JFs refactor.