diff --git a/src/query/base/base-query.controller.ts b/src/query/base/base-query.controller.ts
index 23c2296..859ebf5 100644
--- a/src/query/base/base-query.controller.ts
+++ b/src/query/base/base-query.controller.ts
@@ -36,7 +36,7 @@ export abstract class BaseQueryController {
   @Post("create")
   async createQuery(
     @Req() req: Request & { apiToken: Token },
-    @Body() queryData: { source: string }
+    @Body() queryData: { source: string; isTypescript?: number }
   ) {
     return this.queryHandlerService.createQuery(
       {
@@ -50,7 +50,7 @@ export abstract class BaseQueryController {
   @Post("update/:id")
   @UseGuards(QueryGuard)
   async updateQuery(
-    @Body() updateData: Partial<{ source: string }>,
+    @Body() updateData: Partial<{ source: string; isTypescript?: number }>,
     @Param("id") id: string
   ) {
     return this.queryHandlerService.updateQuery(id, updateData);
diff --git a/src/query/handler/query.handler.service.ts b/src/query/handler/query.handler.service.ts
index 5a375ea..0373cf2 100644
--- a/src/query/handler/query.handler.service.ts
+++ b/src/query/handler/query.handler.service.ts
@@ -31,7 +31,6 @@ export class QueryHandlerService {
     delete queryData.projectToken;
 
     const query = this.queryRepository.create(queryData);
-
     await this.queryRepository.save(query);
 
     return query;
@@ -64,9 +63,9 @@ export class QueryHandlerService {
     }
 
     await this.redisClient.del(`query_${id}`);
-
     Object.assign(query, updateData);
-    return this.queryRepository.save(query);
+
+    return await this.queryRepository.save(query);
   }
 
   async deleteQuery(id: string) {