minor test fix, with basic mocking in place