- Fix missing return