diff --git a/ts/flexbuffers.ts b/ts/flexbuffers.ts
index 40482dc..a59d497 100644
--- a/ts/flexbuffers.ts
+++ b/ts/flexbuffers.ts
@@ -1,13 +1,14 @@
 /* eslint-disable @typescript-eslint/no-namespace */
-import { Builder } from './flexbuffers/builder'
-import { toReference as toReferenceFunction } from './flexbuffers/reference';
+import { Builder } from './flexbuffers/builder.js'
+import { toReference } from './flexbuffers/reference.js'
+export { toReference } from './flexbuffers/reference.js'
 
 export function builder(): Builder {
     return new Builder();
 }
 
-export function toObject(buffer: Uint8Array): unknown {
-    return toReferenceFunction(buffer).toObject();
+export function toObject(buffer: ArrayBuffer): unknown {
+    return toReference(buffer).toObject();
 }
 
 export function encode(object: unknown, size = 2048, deduplicateStrings = true, deduplicateKeys = true, deduplicateKeyVectors = true): Uint8Array {
@@ -15,16 +16,3 @@
     builder.add(object);
     return builder.finish();
 }
-
-const builderFunction = builder
-const toObjectFunction = toObject
-const encodeFunction = encode
-
-export namespace flexbuffers {
-    export const builder = builderFunction;
-    export const toObject = toObjectFunction;
-    export const encode = encodeFunction;
-    export const toReference = toReferenceFunction;
-}
-
-export default flexbuffers;