diff --git a/pg-query-stream/pg-query-stream-tests.ts b/pg-query-stream/pg-query-stream-tests.ts
index 6c4c034102..04796580de 100644
--- a/pg-query-stream/pg-query-stream-tests.ts
+++ b/pg-query-stream/pg-query-stream-tests.ts
@@ -10,10 +10,10 @@ const query = new QueryStream('SELECT * FROM generate_series(0, $1) num', [10000
 
 pg.connect('', (err, client, done) => {
     const stream = client.query(query);
-    stream.on('end', () => {
+    stream.then(() => {
         client.end();
     });
-    stream.on('data', (data: any) => {
+    stream.then((data: any) => {
         console.log(data);
     });
 });