viewport improvements

1 files changed, 1 insertions, 1 deletions

diff --git a/src/imgui_ui_stable_diffusion.cpp b/src/imgui_ui_stable_diffusion.cpp
index 9f2963a..6febdf3 100644
--- a/src/imgui_ui_stable_diffusion.cpp
+++ b/src/imgui_ui_stable_diffusion.cpp
@@ -98,7 +98,7 @@ ImGui_SD_Prompt(project_data *File, project_state *State, ui *UI, memory *Memory
                     sorted_layer_array SortEntry = SortedLayerStart[i];
                     uint32 Index_Physical = SortEntry.Block_Layer_Index;
                     block_layer *TestLayer = (block_layer *)Memory_Block_AddressAtIndex(Memory, F_Layers, Index_Physical);
-                    if (TestLayer->IsSelected) {
+                    if (TestLayer->IsSelected & 0x01) {
                         Layer = TestLayer;
                         break;
                     }