diff --git a/PVEDiscordDark/js/PVEDiscordDark.js b/PVEDiscordDark/js/PVEDiscordDark.js
index eaa37cf..ec6bd50 100644
--- a/PVEDiscordDark/js/PVEDiscordDark.js
+++ b/PVEDiscordDark/js/PVEDiscordDark.js
@@ -41,7 +41,8 @@ function patchBackupConfig() {
 };
 
 function patchDiskSmartWindow() {
-	PVE.DiskSmartWindow.prototype.items[1].style['background-color'] = '#23272a';
+	const target = PVE.DiskSmartWindow || Proxmox.window.DiskSmart;
+	target.prototype.items[1].style['background-color'] = '#23272a';
 }
 
 function patchTFAEdit() {
@@ -185,4 +186,4 @@ patchDiskSmartWindow();
 patchTFAEdit();
 patchSummary();
 patchSubscription();
-console.log('PVEDiscordDark :: Patched');
\ No newline at end of file
+console.log('PVEDiscordDark :: Patched');