try to use mock instead of stub