add missing return statement. I am stupid.