diff --git a/CustomImplementations/FGColoredInstanceMeshProxy.cpp b/CustomImplementations/FGColoredInstanceMeshProxy.cpp
index 1f21645a..52167872 100644
--- a/CustomImplementations/FGColoredInstanceMeshProxy.cpp
+++ b/CustomImplementations/FGColoredInstanceMeshProxy.cpp
@@ -1,6 +1,6 @@
 bool UFGColoredInstanceMeshProxy::ShouldCreateRenderState() const {
-#if WITH_EDITOR // Ensure visibility in blueprint & editor.
-    if (GetWorld()->WorldType == EWorldType::Editor || GetWorld()->WorldType == EWorldType::EditorPreview) {
+#if WITH_EDITOR // Ensure visibility in blueprint, editor & PIE
+    if (GetWorld()->IsEditorWorld()) {
         return true;
     }
 #endif