diff --git a/lib/pycriu/images/__init__.py b/lib/pycriu/images/__init__.py
index ea87e4e7b..12292cce3 100644
--- a/lib/pycriu/images/__init__.py
+++ b/lib/pycriu/images/__init__.py
@@ -1,5 +1,6 @@
 import sys, os
 sys.path.append(os.path.dirname(os.path.realpath(__file__)))
 from .magic import *
-from .images import *
+from .images import (load, loads, dump, dumps, info,
+                     handlers, MagicException, entry_handler)
 from .pb import *