Modified the test to use test double.