diff --git a/Makefile b/Makefile
index 1824ea180..05834d682 100644
--- a/Makefile
+++ b/Makefile
@@ -452,6 +452,8 @@ ruff:
 		test/inhfd/*.py \
 		test/others/rpc/config_file.py \
 		test/others/action-script/check_actions.py \
+		lib/pycriu/criu.py \
+		lib/pycriu/__init__.py \
 		lib/pycriu/images/pb2dict.py \
 		lib/pycriu/images/images.py \
 		scripts/criu-ns \
diff --git a/lib/pycriu/__init__.py b/lib/pycriu/__init__.py
index 2abcf029d..28f1e9424 100644
--- a/lib/pycriu/__init__.py
+++ b/lib/pycriu/__init__.py
@@ -1,4 +1,15 @@
 from . import rpc_pb2 as rpc
 from . import images
-from .criu import *
-from .version import __version__
\ No newline at end of file
+from .criu import criu, CRIUExceptionExternal, CRIUException
+from .criu import CR_DEFAULT_SERVICE_ADDRESS
+from .version import __version__
+
+__all__ = (
+    "rpc",
+    "images",
+    "criu",
+    "CRIUExceptionExternal",
+    "CRIUException",
+    "CR_DEFAULT_SERVICE_ADDRESS",
+    "__version__",
+)
\ No newline at end of file