diff --git a/pmb/export/odin.py b/pmb/export/odin.py
index a0fce30f..774dea39 100644
--- a/pmb/export/odin.py
+++ b/pmb/export/odin.py
@@ -42,8 +42,8 @@ def odin(args, flavor, folder):
                            " Only 'heimdall' methods are supported.")
 
     # Partitions
-    partition_kernel = args.deviceinfo["flash_heimdall_partition_kernel"]
-    partition_initfs = args.deviceinfo["flash_heimdall_partition_initfs"]
+    partition_kernel = args.deviceinfo["flash_heimdall_partition_kernel"] or "KERNEL"
+    partition_initfs = args.deviceinfo["flash_heimdall_partition_initfs"] or "RECOVERY"
 
     # Temporary folder
     temp_folder = "/tmp/odin-flashable-tar"