Thanks :). I didn't know how to make it a function since it takes values from `self.returns` which can be changed by test case.