diff --git a/src/crypto.ts b/src/crypto.ts
index 0d33d33..5d583d2 100644
--- a/src/crypto.ts
+++ b/src/crypto.ts
@@ -2,7 +2,7 @@ import { Handshake } from "./handshake";
 import { Buffer } from "buffer";
 
 interface ReturnEncryptionWrapper {
-  (source: Iterable<Uint8Array>): any;
+  (source: Iterable<Uint8Array>): AsyncIterableIterator<Uint8Array>;
 }
 
 // Returns generator that encrypts payload from the user