add back missing return