Add the right css to the CurrentScreen block

nodeEditor/src/diagram/display/textureDisplayManager.ts

@@ -4,6 +4,7 @@ import { TextureBlock } from 'babylonjs/Materials/Node/Blocks/Dual/textureBlock'
 import { RefractionBlock } from 'babylonjs/Materials/Node/Blocks/PBR/refractionBlock';
 import { ReflectionTextureBlock } from 'babylonjs/Materials/Node/Blocks/Dual/reflectionTextureBlock';
 import { TextureLineComponent } from '../../sharedComponents/textureLineComponent';
+import { CurrentScreenBlock } from 'babylonjs/Materials/Node/Blocks/Dual/currentScreenBlock';
 
 export class TextureDisplayManager implements IDisplayManager {
     private _previewCanvas: HTMLCanvasElement;
@@ -30,7 +31,7 @@ export class TextureDisplayManager implements IDisplayManager {
 
         if (!this._previewCanvas) {
             contentArea.classList.add("texture-block");
-            if (block instanceof TextureBlock) {
+            if (block instanceof TextureBlock || block instanceof CurrentScreenBlock) {
                 contentArea.classList.add("regular-texture-block");
             }