Overriding for test