are you suggesting I use a mock instead of the function?