fix up mock for test