Fix parsing of arrow functions