fix 'return'