diff --git a/www/apps/book/app/learn/advanced-development/modules/db-operations/page.mdx b/www/apps/book/app/learn/advanced-development/modules/db-operations/page.mdx
index 481ee8f993f23..3431ed1ba41b2 100644
--- a/www/apps/book/app/learn/advanced-development/modules/db-operations/page.mdx
+++ b/www/apps/book/app/learn/advanced-development/modules/db-operations/page.mdx
@@ -28,8 +28,8 @@ So, to run database queries in a service:
 For example, in your service, add the following methods:
 
 export const methodsHighlight = [
-  ["11", "getCount", "Retrieves the number of records in `my_custom` using the `count` method."],
-  ["18", "getCountSql", "Retrieves the number of records in `my_custom` using the `execute` method."]
+  ["12", "getCount", "Retrieves the number of records in `my_custom` using the `count` method."],
+  ["19", "getCountSql", "Retrieves the number of records in `my_custom` using the `execute` method."]
 ]
 
 ```ts highlights={methodsHighlight}
@@ -38,6 +38,7 @@ import {
   InjectManager,
   MedusaContext,
 } from "@medusajs/framework/utils"
+import { SqlEntityManager } from "@mikro-orm/knex"
 
 class HelloModuleService {
   // ...