From fc6897d7c2b33b4b42db6ff8eca0f907086cb00f Mon Sep 17 00:00:00 2001
From: Arthur <worldnomad@gmail.com>
Date: Thu, 19 Oct 2023 15:51:48 -0700
Subject: [PATCH] 8356 cleanup

---
 netbox/virtualization/forms/filtersets.py       | 2 +-
 netbox/virtualization/tables/virtualmachines.py | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/netbox/virtualization/forms/filtersets.py b/netbox/virtualization/forms/filtersets.py
index 1a2b63ba5..bc3160e8b 100644
--- a/netbox/virtualization/forms/filtersets.py
+++ b/netbox/virtualization/forms/filtersets.py
@@ -228,7 +228,7 @@ class VirtualDiskFilterForm(NetBoxModelFilterSetForm):
     model = VirtualDisk
     fieldsets = (
         (None, ('q', 'filter_id', 'tag')),
-        # (_('Virtual Machine'), ('virtual_machine_id')),
+        (_('Virtual Machine'), ('virtual_machine_id',)),
     )
     virtual_machine_id = DynamicModelMultipleChoiceField(
         queryset=VirtualMachine.objects.all(),
diff --git a/netbox/virtualization/tables/virtualmachines.py b/netbox/virtualization/tables/virtualmachines.py
index aab7e997a..e1ee708bf 100644
--- a/netbox/virtualization/tables/virtualmachines.py
+++ b/netbox/virtualization/tables/virtualmachines.py
@@ -178,9 +178,9 @@ class VirtualDiskTable(NetBoxTable):
     class Meta(NetBoxTable.Meta):
         model = VirtualDisk
         fields = (
-            'pk', 'id', 'name', 'size', 'tags',
+            'pk', 'id', 'virtual_machine', 'name', 'size', 'tags',
         )
-        default_columns = ('pk', 'name', 'size')
+        default_columns = ('pk', 'name', 'virtual_machine', 'size')
         row_attrs = {
             'data-name': lambda record: record.name,
         }