Covering all code

2 files changed, 3 insertions, 6 deletions

diff --git a/src/edify/main.py b/src/edify/main.py
index 78d5561..ab326ce 100644
--- a/src/edify/main.py
+++ b/src/edify/main.py
@@ -1,4 +1,4 @@
 # Edify Package
 
 def main():
-    pass
+    return 0
diff --git a/tests/test_edify.py b/tests/test_edify.py
index 782c347..7dcadb9 100644
--- a/tests/test_edify.py
+++ b/tests/test_edify.py
@@ -1,7 +1,4 @@
-
-from edify import main
-
+from edify.main import main
 
 def test_main():
-    if main.main() is None:
-        assert True
+    assert main() == 0