Fixed return statements by not dropping them from the AST.