Fixed unit test/coverage for server mode.