2 files changed, 9 insertions, 1 deletions
diff --git a/lib/ManaTools/Shared/disk_backend/Plugin/Disk.pm b/lib/ManaTools/Shared/disk_backend/Plugin/Disk.pm
index 2b3de942..616bcb0e 100644
--- a/lib/ManaTools/Shared/disk_backend/Plugin/Disk.pm
+++ b/lib/ManaTools/Shared/disk_backend/Plugin/Disk.pm
@@ -112,7 +112,11 @@ extends 'ManaTools::Shared::disk_backend::Part';
 use MooseX::ClassAttribute;
 
 class_has '+type' => (
-    default => 'Disks'
+    default => 'Disks',
+);
+
+class_has '+childVisualization' => (
+    default => ManaTools::Shared::Visualization->tabVisualization,
 );
 
 class_has '+restrictions' => (
diff --git a/lib/ManaTools/Shared/disk_backend/Plugin/Partition.pm b/lib/ManaTools/Shared/disk_backend/Plugin/Partition.pm
index 534beda0..cc37f7e9 100644
--- a/lib/ManaTools/Shared/disk_backend/Plugin/Partition.pm
+++ b/lib/ManaTools/Shared/disk_backend/Plugin/Partition.pm
@@ -218,6 +218,10 @@ class_has '+type' => (
     default => 'PartitionTable'
 );
 
+class_has '+childVisualization' => (
+    default => ManaTools::Shared::Visualization->buttonboxVisualization,
+);
+
 class_has '+restrictions' => (
     default => sub {
         return {