diff --git a/api/template/service.js b/api/template/service.js
index 1bcae567540a31315475064f4acd1cc2771dc49c..e1ac2b12614ba8e4f156aeef1d3e24e35bf694ea 100644
--- a/api/template/service.js
+++ b/api/template/service.js
@@ -60,7 +60,7 @@ const getParticularTemplate = async (page,platform) => {
   try {
     const template = await Template.findOne({ name: page,platform}).populate('widgets.widgetId');
     if (!template) {
-      return res.status(404).json({ error: 'Template not found' });
+      return { status: 404,  message: 'Template not found'};
     }
     return template;
   } catch (error) {
diff --git a/swagger.yaml b/swagger.yaml
index 63757fda8507a69fba3cbf4baa84384bdff69ff5..f85f05fa9cad3b146033c5297ad55637c0601931 100644
--- a/swagger.yaml
+++ b/swagger.yaml
@@ -21,12 +21,13 @@ paths:
           schema:
             type: string
           description: The page for which template details are requested.
-        - in: path
+        - in: query # Add platform as a query parameter
           name: platform
           required: true
           schema:
             type: string
-          description: The platform for which template details are requested.
+            enum: ['web', 'mobile', 'both']
+          description: The platform for which template details are requested (web, mobile, or both).
       responses:
         '200':
           description: Template retrieved successfully