diff --git a/apps/contrast-plugin/src/app/app.component.ts b/apps/contrast-plugin/src/app/app.component.ts
index 0ce0b73..f67d970 100644
--- a/apps/contrast-plugin/src/app/app.component.ts
+++ b/apps/contrast-plugin/src/app/app.component.ts
@@ -13,7 +13,7 @@ import type {
 } from '../model';
 import { filter, fromEvent, map, merge, take } from 'rxjs';
 import { CommonModule } from '@angular/common';
-import { PenpotShape } from '@penpot/plugin-types';
+import { Shape } from '@penpot/plugin-types';
 
 @Component({
   standalone: true,
@@ -221,7 +221,7 @@ export class AppComponent {
     return [r, g, b];
   }
 
-  #getShapeColor(shape?: PenpotShape): string | undefined {
+  #getShapeColor(shape?: Shape): string | undefined {
     const fills = shape?.fills;
     if (fills && fills !== 'mixed') {
       return fills?.[0]?.fillColor ?? shape?.strokes?.[0]?.strokeColor;
diff --git a/apps/contrast-plugin/src/model.ts b/apps/contrast-plugin/src/model.ts
index 214a79e..37de33b 100644
--- a/apps/contrast-plugin/src/model.ts
+++ b/apps/contrast-plugin/src/model.ts
@@ -1,4 +1,4 @@
-import { PenpotShape } from '@penpot/plugin-types';
+import { Shape } from '@penpot/plugin-types';
 
 export interface InitPluginUIEvent {
   type: 'ready';
@@ -10,12 +10,12 @@ export interface InitPluginEvent {
   type: 'init';
   content: {
     theme: string;
-    selection: PenpotShape[];
+    selection: Shape[];
   };
 }
 export interface SelectionPluginEvent {
   type: 'selection';
-  content: PenpotShape[];
+  content: Shape[];
 }
 
 export interface ThemePluginEvent {