Fixed mock tests.