diff --git a/src/space-model/services/space-model-product-allocation.service.ts b/src/space-model/services/space-model-product-allocation.service.ts
index bd8c5e9..56bbb24 100644
--- a/src/space-model/services/space-model-product-allocation.service.ts
+++ b/src/space-model/services/space-model-product-allocation.service.ts
@@ -1,4 +1,4 @@
-import { QueryRunner } from 'typeorm';
+import { In, QueryRunner } from 'typeorm';
 import { HttpException, HttpStatus, Injectable } from '@nestjs/common';
 
 import {
diff --git a/src/space-model/space-model.module.ts b/src/space-model/space-model.module.ts
index e2aaaaa..5b15275 100644
--- a/src/space-model/space-model.module.ts
+++ b/src/space-model/space-model.module.ts
@@ -42,6 +42,8 @@ import { TuyaService } from '@app/common/integrations/tuya/services/tuya.service
 import { CommunityRepository } from '@app/common/modules/community/repositories';
 import { TagService as NewTagService } from 'src/tags/services/tags.service';
 import { NewTagRepository } from '@app/common/modules/tag/repositories/tag-repository';
+import { SpaceModelProductAllocationService } from './services/space-model-product-allocation.service';
+import { SubspaceModelProductAllocationService } from './services/subspace/subspace-model-product-allocation.service';
 
 const CommandHandlers = [
   PropogateUpdateSpaceModelHandler,
@@ -80,6 +82,9 @@ const CommandHandlers = [
     SpaceModelProductAllocationRepoitory,
     SubspaceModelProductAllocationRepoitory,
     NewTagRepository,
+    SpaceModelProductAllocationService,
+    SubspaceModelProductAllocationRepoitory,
+    SubspaceModelProductAllocationService,
   ],
   exports: [CqrsModule, SpaceModelService],
 })
diff --git a/src/space/space.module.ts b/src/space/space.module.ts
index ba7df62..a4f904d 100644
--- a/src/space/space.module.ts
+++ b/src/space/space.module.ts
@@ -75,6 +75,8 @@ import {
 import { AutomationRepository } from '@app/common/modules/automation/repositories';
 import { TagService as NewTagService } from 'src/tags/services/tags.service';
 import { NewTagRepository } from '@app/common/modules/tag/repositories/tag-repository';
+import { SpaceModelProductAllocationService } from 'src/space-model/services/space-model-product-allocation.service';
+import { SubspaceModelProductAllocationService } from 'src/space-model/services/subspace/subspace-model-product-allocation.service';
 
 export const CommandHandlers = [DisableSpaceHandler];
 
@@ -139,6 +141,8 @@ export const CommandHandlers = [DisableSpaceHandler];
     SpaceModelProductAllocationRepoitory,
     SubspaceModelProductAllocationRepoitory,
     NewTagRepository,
+    SpaceModelProductAllocationService,
+    SubspaceModelProductAllocationService,
   ],
   exports: [SpaceService],
 })