Fixup missing return