Add back missing return statement.