1 files changed, 10 insertions, 3 deletions
diff --git a/frontend/src/app/_elements/model-load/model-load.component.ts b/frontend/src/app/_elements/model-load/model-load.component.ts
index dbca3d17..4541c114 100644
--- a/frontend/src/app/_elements/model-load/model-load.component.ts
+++ b/frontend/src/app/_elements/model-load/model-load.component.ts
@@ -38,9 +38,16 @@ export class ModelLoadComponent implements OnInit {
   showMyModels: boolean = true;
 
   constructor(private modelsService: ModelsService) {
-    this.modelsService.getMyModels().subscribe((models) => {
-      this.myModels = models;
-    });
+    if (this.forExperiment == undefined) {
+      this.modelsService.getMyModels().subscribe((models) => {
+        this.myModels = models;
+      });
+    }
+    else {
+      this.modelsService.getMyModelsByType(this.forExperiment.type).subscribe((models) => {
+        this.myModels = models;
+      });
+    }
   }
 
   ngOnInit(): void {