Try using mock instead