diff --git a/ui/components/canvas2d/src/api/boundingBoxesApi.ts b/ui/components/canvas2d/src/api/boundingBoxesApi.ts
index c8d929d23..a9bdede83 100644
--- a/ui/components/canvas2d/src/api/boundingBoxesApi.ts
+++ b/ui/components/canvas2d/src/api/boundingBoxesApi.ts
@@ -1,7 +1,7 @@
+import type { BBox, Mask, SelectionTool } from "@pixano/core";
+import LockIcon from "@pixano/core/src/assets/icons/lockIcon.svg";
 import Konva from "konva";
 import simplify from "simplify-js";
-import type { Mask, BBox, SelectionTool } from "@pixano/core";
-import LockIcon from "@pixano/core/src/assets/icons/lockIcon.svg";
 
 import { BBOX_STROKEWIDTH, MASK_STROKEWIDTH } from "../lib/constants";
 import type { PolygonGroupDetails, PolygonGroupPoint } from "../lib/types/canvas2dTypes";
@@ -174,6 +174,13 @@ export function addBBox(
   bboxGroup.add(bboxKonva);
 
   bboxRect.on("transform", function () {
+    const rect: Konva.Rect = bboxKonva.findOne(`#rect${bbox.id}`);
+    rect.setAttrs({
+      width: rect.width() * rect.scaleX(),
+      height: rect.height() * rect.scaleY(),
+      scaleX: 1,
+      scaleY: 1,
+    });
     stickLabelsToRectangle(tooltip, lockTooltip, bboxRect);
   });
 
diff --git a/ui/components/imageWorkspace/src/components/ActionsTabs/ObjectTabModelContent.svelte b/ui/components/imageWorkspace/src/components/ActionsTabs/ObjectTabModelContent.svelte
index d37b2e661..23ecd867b 100644
--- a/ui/components/imageWorkspace/src/components/ActionsTabs/ObjectTabModelContent.svelte
+++ b/ui/components/imageWorkspace/src/components/ActionsTabs/ObjectTabModelContent.svelte
@@ -13,13 +13,13 @@
    *
    * http://www.cecill.info
    */
-  import { Eye, EyeOff, ChevronRight } from "lucide-svelte";
+  import { ChevronRight, Eye, EyeOff } from "lucide-svelte";
 
   import { cn, IconButton } from "@pixano/core/src";
 
-  import { itemObjects } from "../../lib/stores/imageWorkspaceStores";
   import { toggleObjectDisplayControl } from "../../lib/api/objectsApi";
   import { GROUND_TRUTH, MODEL_RUN } from "../../lib/constants";
+  import { itemObjects } from "../../lib/stores/imageWorkspaceStores";
 
   export let sectionTitle: string;
   export let modelName: string;
@@ -29,6 +29,7 @@
   let open: boolean = true;
 
   itemObjects.subscribe((items) => {
+    if (!items.length) return;
     const allObjectsOfCurrentModel = items.filter((item) => {
       if (modelName === MODEL_RUN) {
         return item.source_id !== GROUND_TRUTH;
diff --git a/ui/components/imageWorkspace/src/components/Toolbar.svelte b/ui/components/imageWorkspace/src/components/Toolbar.svelte
index fec936bad..8b5f27cfc 100644
--- a/ui/components/imageWorkspace/src/components/Toolbar.svelte
+++ b/ui/components/imageWorkspace/src/components/Toolbar.svelte
@@ -105,7 +105,7 @@
   </div>
   <div
     class={cn("flex items-center flex-col gap-4 mt-4", {
-      "bg-slate-100 rounded-sm": showSmartTools,
+      "bg-slate-200 rounded-sm": showSmartTools,
     })}
   >
     <button