diff --git a/setup-wizard/src/app/modals/cifs-modal/cifs-modal.page.ts b/setup-wizard/src/app/modals/cifs-modal/cifs-modal.page.ts
index b8bc93731..4b0ded99b 100644
--- a/setup-wizard/src/app/modals/cifs-modal/cifs-modal.page.ts
+++ b/setup-wizard/src/app/modals/cifs-modal/cifs-modal.page.ts
@@ -10,6 +10,7 @@ import { PasswordPage } from '../password/password.page'
 })
 export class CifsModal {
   cifs = {
+    type: 'cifs' as 'cifs',
     hostname: '',
     path: '',
     username: '',
diff --git a/setup-wizard/src/app/pages/recover/recover.page.ts b/setup-wizard/src/app/pages/recover/recover.page.ts
index 72cb08061..3a64a796e 100644
--- a/setup-wizard/src/app/pages/recover/recover.page.ts
+++ b/setup-wizard/src/app/pages/recover/recover.page.ts
@@ -102,6 +102,7 @@ export class RecoverPage {
       if (res.role === 'success') {
         const { hostname, path, username, password } = res.data.cifs
         this.stateService.recoverySource = {
+          type: 'cifs',
           hostname,
           path,
           username,
@@ -131,7 +132,7 @@ export class RecoverPage {
         }
       })
       await modal.present()
-    // if no product key, it means they are an upgrade kit user
+      // if no product key, it means they are an upgrade kit user
     } else {
       const modal = await this.modalController.create({
         component: ProdKeyModal,
@@ -165,6 +166,7 @@ export class RecoverPage {
 
   private async selectRecoverySource (logicalname: string, password?: string) {
     this.stateService.recoverySource = {
+      type: 'disk',
       logicalname,
     }
     this.stateService.recoveryPassword = password
diff --git a/setup-wizard/src/app/services/api/api.service.ts b/setup-wizard/src/app/services/api/api.service.ts
index e17a24405..cb6f5b7eb 100644
--- a/setup-wizard/src/app/services/api/api.service.ts
+++ b/setup-wizard/src/app/services/api/api.service.ts
@@ -57,10 +57,12 @@ export interface CifsBackupTarget {
 }
 
 export interface DiskRecoverySource {
+  type: 'disk'
   logicalname: string // partition logicalname
 }
 
 export interface CifsRecoverySource {
+  type: 'cifs'
   hostname: string
   path: string
   username: string