diff --git a/__init__.py b/__init__.py
index a3a3218..c5b1f09 100644
--- a/__init__.py
+++ b/__init__.py
@@ -1,3 +1,3 @@
 __doc__ = __doc__ or ""
 
-__all__ = ["util","fileserver","xss_handler"]
+__all__ = ["util","fileserver","xss_handler","genRevShell"]
diff --git a/fileserver.py b/fileserver.py
index 76114be..81d29bf 100755
--- a/fileserver.py
+++ b/fileserver.py
@@ -1,8 +1,7 @@
 #!/usr/bin/env python
 
-from hackingscripts import util
+from hackingscripts import util, xss_handler
 from http.server import BaseHTTPRequestHandler, HTTPServer
-from . import xss_handler
 import threading
 import requests
 import sys
diff --git a/genRevShell.py b/genRevShell.py
index bd6799c..1681b79 100755
--- a/genRevShell.py
+++ b/genRevShell.py
@@ -2,7 +2,7 @@
 
 import socket
 import sys
-from . import util
+import util
 import pty
 
 def generatePayload(type, local_address, port):
diff --git a/update.sh b/update.sh
index 91730ac..7bd94ca 100755
--- a/update.sh
+++ b/update.sh
@@ -23,6 +23,7 @@ location=$(curl -s -I https://github.com/jpillora/chisel/releases/latest | grep
 if [[ "$location" =~ ^https://github.com/jpillora/chisel/releases/tag/v(.*) ]]; then
   chisel_version=${BASH_REMATCH[1]}
   chisel_version=${chisel_version%%[[:space:]]}
+  echo "Got Chisel version: ${chisel_version}"
   curl -s -L "https://github.com/jpillora/chisel/releases/download/v${chisel_version}/chisel_${chisel_version}_linux_386.gz"    | gzip -d > chisel
   curl -s -L "https://github.com/jpillora/chisel/releases/download/v${chisel_version}/chisel_${chisel_version}_linux_amd64.gz"  | gzip -d > chisel64
   curl -s -L "https://github.com/jpillora/chisel/releases/download/v${chisel_version}/chisel_${chisel_version}_windows_386.gz"  | gzip -d > win/chisel.exe