Add missing 'return'