Add support for mock config flag.