diff --git a/abi_markdowner/main.py b/abi_markdowner/main.py
index 740febb..4804dd5 100644
--- a/abi_markdowner/main.py
+++ b/abi_markdowner/main.py
@@ -3,8 +3,8 @@
 import json
 import toml
 import argparse
-from abi_markdowner import generate_markdown_from_abi
-from file_io import read_abi_from_file, save_markdown_to_file
+from abi_markdowner.abi_markdowner import generate_markdown_from_abi
+from abi_markdowner.file_io import read_abi_from_file, save_markdown_to_file
 
 # Default values
 SC_PATH = './'