add missing return statement