diff --git a/mitmproxy2swagger/mitmproxy2swagger.py b/mitmproxy2swagger/mitmproxy2swagger.py
index df42dbd..4206dd6 100755
--- a/mitmproxy2swagger/mitmproxy2swagger.py
+++ b/mitmproxy2swagger/mitmproxy2swagger.py
@@ -13,10 +13,10 @@
 import argparse
 import ruamel.yaml
 import re
-import swagger_util
-from har_capture_reader import HarCaptureReader, har_archive_heuristic
-from mitmproxy_capture_reader import MitmproxyCaptureReader, mitmproxy_dump_file_huristic
-import console_util
+import mitmproxy2swagger.console_util
+from mitmproxy2swagger.har_capture_reader import HarCaptureReader, har_archive_heuristic
+from mitmproxy2swagger.mitmproxy_capture_reader import MitmproxyCaptureReader, mitmproxy_dump_file_huristic
+import mitmproxy2swagger.console_util
 
 def path_to_regex(path):
     # replace the path template with a regex
diff --git a/pyproject.toml b/pyproject.toml
index fddcdf3..b1a60df 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "mitmproxy2swagger"
-version = "0.5.0"
+version = "0.5.1"
 description = ""
 authors = ["alufers <alufers@wp.pl>"]
 readme = "README.md"