Add missing `return`