diff --git a/src/dynamic_graph/sot/core/__init__.py b/src/dynamic_graph/sot/core/__init__.py
index b833e8d6..0a05883f 100644
--- a/src/dynamic_graph/sot/core/__init__.py
+++ b/src/dynamic_graph/sot/core/__init__.py
@@ -1,2 +1,2 @@
 from . import wrap  # noqa
-from .wrap import Flags  # noqa
+from .wrap import Device, Flags  # noqa