Adding missing return statement