diff --git a/common/cmd_ubi.c b/common/cmd_ubi.c
index 394a169a4a..e00dd66d55 100644
--- a/common/cmd_ubi.c
+++ b/common/cmd_ubi.c
@@ -486,6 +486,8 @@ int ubi_part(char *part_name, const char *vid_header_offset)
 	if (ubi_initialized) {
 		ubi_exit();
 		del_mtd_partitions(ubi_dev.mtd_info);
+		put_mtd_device(ubi_dev.mtd_info);
+		ubi_initialized = 0;
 	}
 
 	/*