diff --git a/playbooks/mdstat.yml b/playbooks/mdstat.yml
index d1df016ad2135bbb2ef158aadec817e44a94d4b6..01335f1f41feff301d7d09cb40c27286da5ff6b5 100644
--- a/playbooks/mdstat.yml
+++ b/playbooks/mdstat.yml
@@ -11,19 +11,19 @@
   become: yes
   tasks:
     - name: Write to action file
-      shell: echo 'repair' >/sys/block/md2/md/sync_action
+      shell: echo 'repair' >/sys/block/{{ partition }}/md/sync_action
     - name: Wait for repair to complete
       wait_for:
-        path: /sys/block/md2/md/sync_action
+        path: /sys/block/{{ partition }}/md/sync_action
         search_regex: idle
         delay: 60
         sleep: 60
         timeout: 18000
     - name: Write to action file
-      shell: echo 'check' >/sys/block/md2/md/sync_action
+      shell: echo 'check' >/sys/block/{{ partition }}/md/sync_action
     - name: Wait for check to complete
       wait_for:
-        path: /sys/block/md2/md/sync_action
+        path: /sys/block/{{ partition }}/md/sync_action
         search_regex: idle
         delay: 60
         sleep: 60
diff --git a/scripts/mdstat.yml b/scripts/mdstat.yml
index 5ff9f3e0d89c98f1b83e75bd07732b54f5e65dde..731554808e37e648d262fb34f404a47a10ab4b0b 100644
--- a/scripts/mdstat.yml
+++ b/scripts/mdstat.yml
@@ -1,5 +1,15 @@
 ---
 
 description: Run mdstat tasks
+cli:
+  options:
+    - name: partition
+      default: md2
+      help: Which partition to repair
 arguments:
   playbook: mdstat
+command:
+  - name: extra-vars
+    value: partition=%s
+    args:
+      - partition