diff --git a/lib/templateConfigValidator.js b/lib/templateConfigValidator.js
index 0f6e5dc87..26219f0e2 100644
--- a/lib/templateConfigValidator.js
+++ b/lib/templateConfigValidator.js
@@ -30,13 +30,14 @@ module.exports.validateTemplateConfig = (templateConfig, templateParams, asyncap
   isRequiredParamProvided(parameters, templateParams);
   isProvidedTemplateRendererSupported(templateConfig);
   if (asyncapiDocument) {
+    let server;
     if (usesNewAPI(templateConfig)) {
-
+      server = asyncapiDocument.servers().get(templateParams.server);
     } else {
-      const server = asyncapiDocument.servers().get(templateParams.server);
-      isServerProvidedInDocument(server, templateParams.server);
-      isServerProtocolSupported(server, supportedProtocols, templateParams.server);
+      server = asyncapiDocument.servers()[templateParams.server];
     }
+    isServerProvidedInDocument(server, templateParams.server);
+    isServerProtocolSupported(server, supportedProtocols, templateParams.server);
   }
 
   isProvidedParameterSupported(parameters, templateParams);