Fix return from foldr