- Added missing return statement