Added missing return statement