update test to use mock