diff --git a/pyproject.toml b/pyproject.toml
index b0a5512..b5b3165 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -3,7 +3,7 @@ line-length = 120
 
 [tool.poetry]
 name = "yaml-patch"
-version = "0.1.2"
+version = "0.1.3"
 description = "Patch yaml strings"
 readme = "README.md"
 authors = ["Diogo de Campos <campos.ddc@gmail.com>"]
@@ -11,7 +11,7 @@ homepage = "https://github.com/campos-ddc/yaml-patch"
 license = "MIT"
 
 [tool.poetry.scripts]
-yaml-patch = 'main:cli'
+yaml-patch = 'yaml_patch.cli:cli'
 
 [tool.poetry.dependencies]
 python = "^3.7"
diff --git a/main.py b/yaml_patch/cli.py
similarity index 100%
rename from main.py
rename to yaml_patch/cli.py