diff --git a/src/lib/components/DiscoveryGraph.svelte b/src/lib/components/DiscoveryGraph.svelte
index 005e347..a687aca 100644
--- a/src/lib/components/DiscoveryGraph.svelte
+++ b/src/lib/components/DiscoveryGraph.svelte
@@ -159,7 +159,7 @@
 			...nodeStyle('Certificate', Newspaper, overlay),
 			...nodeStyle('Object', Document, overlay),
 			...nodeStyle('NetworkSecurityGroup', ShieldCheck, overlay),
-			...nodeStyle('MachineLearningWorkspace', Beaker, overlay)
+			...nodeStyle('MLWorkspace', Beaker, overlay)
 		]);
 
 		return styles;