From e360dc8649d5ad1df857b942c622aaee4b42956e Mon Sep 17 00:00:00 2001
From: Robert Chu <allegrormc@gmail.com>
Date: Thu, 3 Oct 2024 17:47:01 -0700
Subject: [PATCH] Adds type keyword to exports.

---
 src/compat/waffle.ts       |  2 +-
 src/index.ts               | 17 +++++++++++------
 test/mock-contract.test.ts |  2 +-
 test/waffle-compat.test.ts |  2 +-
 4 files changed, 14 insertions(+), 9 deletions(-)

diff --git a/src/compat/waffle.ts b/src/compat/waffle.ts
index 4422ee9..3e0e0dc 100644
--- a/src/compat/waffle.ts
+++ b/src/compat/waffle.ts
@@ -10,7 +10,7 @@ import {
   deployMock,
   MockCallExpectation,
   MockContractController,
-} from "../mock-contract";
+} from "../mock-contract.js";
 import { PublicClient, WalletClient } from "viem";
 
 export const doppelgangerAbi = [
diff --git a/src/index.ts b/src/index.ts
index 1c9e484..cca471b 100644
--- a/src/index.ts
+++ b/src/index.ts
@@ -1,8 +1,13 @@
 export {
-  MockReadCallExpectation,
-  MockWriteCallExpectation,
-  MockRevertExpectation,
-  MockCallExpectation,
-  MockContractController,
+  type MockReadCallExpectation,
+  type MockWriteCallExpectation,
+  type MockRevertExpectation,
+  type MockCallExpectation,
+  type MockContractController,
   deployMock,
-} from "./mock-contract";
+} from "./mock-contract.js";
+
+export {
+  type MockContract,
+  deployMockContract,
+} from "./compat/waffle.js";
diff --git a/test/mock-contract.test.ts b/test/mock-contract.test.ts
index 49f92e2..3f2be5b 100644
--- a/test/mock-contract.test.ts
+++ b/test/mock-contract.test.ts
@@ -1,6 +1,6 @@
 import { expect } from "chai";
 import { zeroAddress } from "viem";
-import { deployMock } from "../src/mock-contract";
+import { deployMock } from "../src/mock-contract.js";
 import hre from "hardhat";
 import { ExtractAbiFunction } from "abitype";
 
diff --git a/test/waffle-compat.test.ts b/test/waffle-compat.test.ts
index 040787e..8c127de 100644
--- a/test/waffle-compat.test.ts
+++ b/test/waffle-compat.test.ts
@@ -1,5 +1,5 @@
 import { expect } from "chai";
-import { deployMockContract } from "../src/compat/waffle";
+import { deployMockContract } from "../src/compat/waffle.js";
 import hre from "hardhat";
 import { zeroAddress } from "viem";