1 files changed, 8 insertions, 4 deletions
diff --git a/test/tools/getcn b/test/tools/getcn
index 4cc028ca..212977c3 100755
--- a/test/tools/getcn
+++ b/test/tools/getcn
@@ -1,10 +1,14 @@
 #!/usr/bin/env python
-
 import sys
-sys.path.insert(0, "../..")
-from libmproxy import utils
+sys.path.insert(0, "../../")
+from libmproxy import certutils
+
+if len(sys.argv) > 2:
+    port = int(sys.argv[2])
+else:
+    pport = 443
 
-cn, san = utils.get_remote_cn(sys.argv[1], 443)
+cn, san = certutils.get_remote_cn(sys.argv[1], port)
 print cn
 if san:
     for i in san: