[Fix] Added accidentally omitted `return`