diff --git a/linux/storage/disk.sls b/linux/storage/disk.sls
index 55b8920..4fc96e5 100644
--- a/linux/storage/disk.sls
+++ b/linux/storage/disk.sls
@@ -36,9 +36,18 @@
     - module: create_disk_label_{{ disk_name }}
     - pkg: xfsprogs
 
-{%- if partition.get('mkfs') %}
+{% set end_size = end_size + partition.size -%}
 
-{%- if partition.type == "xfs" %}
+{%- endfor %}
+
+probe_partions_{{ disk_name }}:
+  module.run:
+  - name: partition.probe
+  - device: {{ disk_name }}
+
+{%- for partition in disk.get('partitions', []) %}
+
+{%- if partition.get('mkfs') and partition.type == "xfs" %}
 
 mkfs_partition_{{ disk_name }}_{{ loop.index }}:
   module.run:
@@ -50,11 +59,8 @@
 
 {%- endif %}
 
-{%- endif %}
-
-{% set end_size = end_size + partition.size -%}
-
 {%- endfor %}
+
 {%- endfor %}
 
 {%- endif %}