Updated an UT that is IT

Hence I forgot about it, but CI caught it. Removed
mocks as they were doing wrong after the change.