Added mem patch for return statement.