diff --git a/confluent_client/bin/stats b/confluent_client/bin/stats
index 0893fadb..3dc2dfb0 100755
--- a/confluent_client/bin/stats
+++ b/confluent_client/bin/stats
@@ -19,6 +19,7 @@ import argparse
 import base64
 import csv
 import io
+import os
 import numpy as np
 import sys