Add missed `return`