Add missing Return