Fix return expression