diff --git a/partition.sh b/partition.sh
index 22ea5ab..7407360 100644
--- a/partition.sh
+++ b/partition.sh
@@ -51,6 +51,8 @@ echo -e
 echo "This is your new partition table:"
 lsblk | grep "$dev"
 
+echo $dev > /tmp/disk
+
 # Exit.
 break
 done