diff --git a/src/components/tools/BoundingRectangle.vue b/src/components/tools/BoundingRectangle.vue
index 65d786a5e..91dee6849 100644
--- a/src/components/tools/BoundingRectangle.vue
+++ b/src/components/tools/BoundingRectangle.vue
@@ -6,7 +6,7 @@ import { worldToSVG } from '@/src/utils/vtk-helpers';
 import { nonNullable } from '@/src/utils/index';
 import vtkLPSView2DProxy from '@/src/vtk/LPSView2DProxy';
 import vtkBoundingBox from '@kitware/vtk.js/Common/DataModel/BoundingBox';
-import { Bounds, Vector3 } from '@kitware/vtk.js/types';
+import { Bounds, Vector3 } from '@/types';
 import { onVTKEvent } from '@/src/composables/onVTKEvent';
 
 const props = defineProps<{
diff --git a/src/components/tools/polygon/PolygonWidget2D.vue b/src/components/tools/polygon/PolygonWidget2D.vue
index 73a2df77f..2c3f8303d 100644
--- a/src/components/tools/polygon/PolygonWidget2D.vue
+++ b/src/components/tools/polygon/PolygonWidget2D.vue
@@ -25,7 +25,7 @@ import vtkWidgetFactory, {
   vtkPolygonViewWidget as WidgetView,
 } from '@/src/vtk/PolygonWidget';
 import { Maybe } from '@/src/types';
-import { Vector3 } from '@kitware/vtk.js/types';
+import { Vector3 } from '@/types';
 import { useViewStore } from '@/src/store/views';
 import {
   onViewProxyMounted,