fixed compilation
after function not block body stopped being a 'statement'