diff --git a/mithril-infra/main.data-disk.tf b/mithril-infra/main.data-disk.tf
index e7d6096a167..f8f8718aa5e 100644
--- a/mithril-infra/main.data-disk.tf
+++ b/mithril-infra/main.data-disk.tf
@@ -70,7 +70,7 @@ resource "null_resource" "mithril_mount_data_disk" {
       <<-EOT
 set -e
 # Format data disk if necessary
-if sudo blkid /dev/sdb; then 
+if sudo blkid /dev/disk/by-id/google-mithril-data-disk; then 
   echo "Data disk already formatted"
 else 
   # Format data disk