diff --git a/demo/app.jsx b/demo/app.jsx
index a335265..4124ef9 100644
--- a/demo/app.jsx
+++ b/demo/app.jsx
@@ -1,7 +1,7 @@
+import { ReactP5Wrapper } from "@/main.tsx";
 import React, { useCallback, useMemo, useState } from "react";
 import { createRoot } from "react-dom/client";
 
-import { ReactP5Wrapper } from "../src/main.tsx";
 import "./demo.css";
 import { sketch as box } from "./sketches/box";
 import { sketch as plane } from "./sketches/plane";
diff --git a/src/components/ReactP5Wrapper.tsx b/src/components/ReactP5Wrapper.tsx
index 5836583..508c5b5 100644
--- a/src/components/ReactP5Wrapper.tsx
+++ b/src/components/ReactP5Wrapper.tsx
@@ -1,6 +1,8 @@
 import * as React from "react";
 import { propsAreEqual } from "@utils/propsAreEqual";
 
-const ReactP5WrapperGuard = React.lazy(() => import("./ReactP5WrapperGuard"));
+const ReactP5WrapperGuard = React.lazy(
+  () => import("@components/ReactP5WrapperGuard")
+);
 
 export const ReactP5Wrapper = React.memo(ReactP5WrapperGuard, propsAreEqual);
diff --git a/src/components/ReactP5WrapperGuard.tsx b/src/components/ReactP5WrapperGuard.tsx
index 2b4a573..0b11d71 100644
--- a/src/components/ReactP5WrapperGuard.tsx
+++ b/src/components/ReactP5WrapperGuard.tsx
@@ -7,7 +7,7 @@ import { ReactNode } from "react";
 import { FallbackProps } from "react-error-boundary";
 
 const ReactP5WrapperWithSketch = React.lazy(
-  () => import("./ReactP5WrapperWithSketch")
+  () => import("@components/ReactP5WrapperWithSketch")
 );
 
 const ErrorBoundary = React.lazy(() =>
diff --git a/src/main.tsx b/src/main.tsx
index 09379ce..5531cc7 100644
--- a/src/main.tsx
+++ b/src/main.tsx
@@ -1,6 +1,6 @@
-export { ReactP5Wrapper } from "./components/ReactP5Wrapper";
-export { P5WrapperClassName } from "./constants/P5WrapperClassName";
-export { type P5CanvasInstance } from "./contracts/P5CanvasInstance";
-export { type P5WrapperProps } from "./contracts/P5WrapperProps";
-export { type Sketch } from "./contracts/Sketch";
-export { type SketchProps } from "./contracts/SketchProps";
+export { ReactP5Wrapper } from "@components/ReactP5Wrapper";
+export { P5WrapperClassName } from "@constants/P5WrapperClassName";
+export { type P5CanvasInstance } from "@contracts/P5CanvasInstance";
+export { type P5WrapperProps } from "@contracts/P5WrapperProps";
+export { type Sketch } from "@contracts/Sketch";
+export { type SketchProps } from "@contracts/SketchProps";
diff --git a/src/utils/updateCanvasInstance.ts b/src/utils/updateCanvasInstance.ts
index fdafc8e..e8d9259 100644
--- a/src/utils/updateCanvasInstance.ts
+++ b/src/utils/updateCanvasInstance.ts
@@ -2,9 +2,8 @@ import { type CanvasInstanceRef } from "@contracts/CanvasInstanceRef";
 import { type Sketch } from "@contracts/Sketch";
 import { type SketchProps } from "@contracts/SketchProps";
 import { type WrapperRef } from "@contracts/WrapperRef";
-
-import { createCanvasInstance } from "./createCanvasInstance";
-import { removeCanvasInstance } from "./removeCanvasInstance";
+import { createCanvasInstance } from "@utils/createCanvasInstance";
+import { removeCanvasInstance } from "@utils/removeCanvasInstance";
 
 export function updateCanvasInstance<Props extends SketchProps>(
   canvasInstanceRef: CanvasInstanceRef<Props>,
diff --git a/tests/exports.test.tsx b/tests/exports.test.tsx
index 4cf1ead..0c3fb97 100644
--- a/tests/exports.test.tsx
+++ b/tests/exports.test.tsx
@@ -1,8 +1,7 @@
+import { P5WrapperClassName, ReactP5Wrapper } from "@/main";
 import { createElement, isValidElement } from "react";
 import { assert, describe, expect, it, vi } from "vitest";
 
-import { P5WrapperClassName, ReactP5Wrapper } from "../src/main";
-
 describe("Exports", () => {
   describe("P5WrapperClassName", () => {
     it("Exports the css class name used on the wrapper", () => {
diff --git a/tests/utils/removeCanvasInstance.test.ts b/tests/utils/removeCanvasInstance.test.ts
index 2b7d033..cc85836 100644
--- a/tests/utils/removeCanvasInstance.test.ts
+++ b/tests/utils/removeCanvasInstance.test.ts
@@ -1,11 +1,10 @@
+import { SketchProps } from "@/main";
 import { type CanvasInstanceRef } from "@contracts/CanvasInstanceRef";
 import p5 from "@contracts/p5";
 import { removeCanvasInstance } from "@utils/removeCanvasInstance";
 import { createRef } from "react";
 import { describe, expect, it, vi } from "vitest";
 
-import { SketchProps } from "../../src/main";
-
 describe("removeCanvasInstance", () => {
   it("Calls the remove method on the canvas if it exists", () => {
     const instance = new p5(() => {
diff --git a/tests/utils/updateCanvasInstance.test.ts b/tests/utils/updateCanvasInstance.test.ts
index cd2a310..e9a5faf 100644
--- a/tests/utils/updateCanvasInstance.test.ts
+++ b/tests/utils/updateCanvasInstance.test.ts
@@ -1,3 +1,4 @@
+import { SketchProps } from "@/main";
 import { type CanvasInstanceRef } from "@contracts/CanvasInstanceRef";
 import p5 from "@contracts/p5";
 import { type WrapperRef } from "@contracts/WrapperRef";
@@ -6,8 +7,6 @@ import { updateCanvasInstance } from "@utils/updateCanvasInstance";
 import { createRef } from "react";
 import { describe, expect, it, vi } from "vitest";
 
-import { SketchProps } from "../../src/main";
-
 describe("updateCanvasInstance", () => {
   it("Should update a canvas instance to a new version", () => {
     const sketch = vi.fn();