diff --git a/src/modules/models/components/dialogs/dialog-add-metric/dialog-add-metric.component.ts b/src/modules/models/components/dialogs/dialog-add-metric/dialog-add-metric.component.ts
index 5c341316..3b0cb728 100644
--- a/src/modules/models/components/dialogs/dialog-add-metric/dialog-add-metric.component.ts
+++ b/src/modules/models/components/dialogs/dialog-add-metric/dialog-add-metric.component.ts
@@ -40,6 +40,7 @@ export class DialogAddMetricComponent implements OnDestroy, OnInit {
     public metricSpecificationKinds: IMetricSpecifiocationKind[] = [
         {name: 'Kolmogorov-Smirnov', className: 'KSMetricSpec'},
         {name: 'Autoencoder',        className: 'AEMetricSpec'},
+        {name: 'Image Autoencoder',  className: 'ImageAEMetricSpec'},
         {name: 'Random Forest',      className: 'RFMetricSpec'},
         {name: 'GAN',                className: 'GANMetricSpec'},
         {name: 'Latency',            className: 'LatencyMetricSpec'},
@@ -77,6 +78,7 @@ export class DialogAddMetricComponent implements OnDestroy, OnInit {
             tap(([withHealth, kind]) => {
                 switch (kind) {
                     case 'AEMetricSpec':
+                    case 'ImageAEMetricSpec':
                     case 'RFMetricSpec':
                     case 'LatencyMetricSpec':
                         const config = this.form.get('config') as FormGroup;
@@ -93,6 +95,7 @@ export class DialogAddMetricComponent implements OnDestroy, OnInit {
         kindChange.pipe(withLatestFrom(withHealthChange)).subscribe(([kind, withHealth]) => {
             switch (kind) {
                 case 'AEMetricSpec':
+                case 'ImageAEMetricSpec':
                 case 'RFMetricSpec':
                     const x: any = {
                         input: this.fb.control(''),
diff --git a/src/modules/models/components/model-version-monitoring/model-version-monitoring.component.ts b/src/modules/models/components/model-version-monitoring/model-version-monitoring.component.ts
index a809069a..baed063c 100644
--- a/src/modules/models/components/model-version-monitoring/model-version-monitoring.component.ts
+++ b/src/modules/models/components/model-version-monitoring/model-version-monitoring.component.ts
@@ -83,6 +83,7 @@ export class ModelVersionMonitoringComponent implements OnInit, OnDestroy {
             CounterMetricSpec:      ['counter'],
             KSMetricSpec:           ['kolmogorovsmirnov', 'kolmogorovsmirnov_level'],
             AEMetricSpec:           ['autoencoder_reconstruction'],
+            ImageAEMetricSpec:      ['image_autoencoder_reconstructed'],
             RFMetricSpec:           ['randomforest'],
             GANMetricSpec:          ['gan_outlier', 'gan_inlier'],
             LatencyMetricSpec:      ['latency'],