diff --git a/ansibleplaybookgrapher/__init__.py b/ansibleplaybookgrapher/__init__.py
index e69de29b..20c62ea2 100644
--- a/ansibleplaybookgrapher/__init__.py
+++ b/ansibleplaybookgrapher/__init__.py
@@ -0,0 +1,2 @@
+__version__ = "2.2.0.dev0"
+__prog__ = "ansible-playbook-grapher"
diff --git a/ansibleplaybookgrapher/cli.py b/ansibleplaybookgrapher/cli.py
index 779edf10..b6d9d515 100644
--- a/ansibleplaybookgrapher/cli.py
+++ b/ansibleplaybookgrapher/cli.py
@@ -24,7 +24,7 @@
 from ansible.utils.display import Display
 
 from ansibleplaybookgrapher.grapher import Grapher
-from ansibleplaybookgrapher.version import __prog__, __version__
+from ansibleplaybookgrapher import __prog__, __version__
 from ansibleplaybookgrapher.renderer import OPEN_PROTOCOL_HANDLERS
 from ansibleplaybookgrapher.renderer.graphviz import GraphvizRenderer
 from ansibleplaybookgrapher.renderer.mermaid import (
diff --git a/ansibleplaybookgrapher/parser.py b/ansibleplaybookgrapher/parser.py
index 5585f117..97763596 100644
--- a/ansibleplaybookgrapher/parser.py
+++ b/ansibleplaybookgrapher/parser.py
@@ -1,4 +1,4 @@
-# Copyright (C) 2023 Mohamed El Mouctar HAIDARA
+# Copyright (C) 2024 Mohamed El Mouctar HAIDARA
 #
 # This program is free software: you can redistribute it and/or modify
 # it under the terms of the GNU General Public License as published by
diff --git a/ansibleplaybookgrapher/version.py b/ansibleplaybookgrapher/version.py
deleted file mode 100644
index 20c62ea2..00000000
--- a/ansibleplaybookgrapher/version.py
+++ /dev/null
@@ -1,2 +0,0 @@
-__version__ = "2.2.0.dev0"
-__prog__ = "ansible-playbook-grapher"
diff --git a/setup.py b/setup.py
index 56519d97..5560d32d 100644
--- a/setup.py
+++ b/setup.py
@@ -17,7 +17,7 @@
 
 from setuptools import setup, find_packages
 
-from ansibleplaybookgrapher.version import __version__, __prog__
+from ansibleplaybookgrapher import __version__, __prog__
 
 
 def read_requirements(path):
diff --git a/tests/conftest.py b/tests/conftest.py
index 1e940515..c1917a4e 100644
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -3,7 +3,7 @@
 import pytest
 from ansible.plugins.loader import init_plugin_loader
 
-from ansibleplaybookgrapher.version import __prog__
+from ansibleplaybookgrapher import __prog__
 from ansibleplaybookgrapher.cli import PlaybookGrapherCLI
 from tests import INVENTORY_FILE, FIXTURES_DIR
 
diff --git a/tests/test_cli.py b/tests/test_cli.py
index d3b735d4..93b47471 100644
--- a/tests/test_cli.py
+++ b/tests/test_cli.py
@@ -2,7 +2,7 @@
 from ansible.errors import AnsibleOptionsError
 from ansible.release import __version__ as ansible_version
 
-from ansibleplaybookgrapher.version import __prog__, __version__
+from ansibleplaybookgrapher import __prog__, __version__
 from ansibleplaybookgrapher.cli import PlaybookGrapherCLI
 
 
diff --git a/tests/test_graphviz_renderer.py b/tests/test_graphviz_renderer.py
index 02c2f68b..ebb63679 100644
--- a/tests/test_graphviz_renderer.py
+++ b/tests/test_graphviz_renderer.py
@@ -6,7 +6,7 @@
 import pytest
 from pyquery import PyQuery
 
-from ansibleplaybookgrapher.version import __prog__
+from ansibleplaybookgrapher import __prog__
 from ansibleplaybookgrapher.cli import PlaybookGrapherCLI
 from tests import FIXTURES_DIR
 
diff --git a/tests/test_mermaid_renderer.py b/tests/test_mermaid_renderer.py
index e522a1cd..346836aa 100644
--- a/tests/test_mermaid_renderer.py
+++ b/tests/test_mermaid_renderer.py
@@ -3,7 +3,7 @@
 
 import pytest
 
-from ansibleplaybookgrapher.version import __prog__
+from ansibleplaybookgrapher import __prog__
 from ansibleplaybookgrapher.cli import PlaybookGrapherCLI
 from tests import FIXTURES_DIR