Fix for CodeGenVisitor.