- Added missing return here