fix missing return statement