diff --git a/src/hooks/module/useModule.js b/src/hooks/module/useModule.js
index c7f13dc0..0c884608 100644
--- a/src/hooks/module/useModule.js
+++ b/src/hooks/module/useModule.js
@@ -822,10 +822,7 @@ export function useModule() {
 
   const alignModule = (type, surfaceArray) => {
     surfaceArray.forEach((surface) => {
-      const modules = canvas
-        .getObjects()
-        .filter((module) => module.name === POLYGON_TYPE.MODULE)
-        .filter((module) => module.surfaceId === surface.id)
+      const modules = surface.modules
 
       const objects = getObjects()
       let [top, bottom, left, right] = [0, 0, 0, 0]