Same change as above to make the overload work.