diff --git a/tangostationcontrol/tangostationcontrol/devices/unb2.py b/tangostationcontrol/tangostationcontrol/devices/unb2.py
index bf06f4c6ad586eb4b9d6775c8f0fd1643cb20045..5b433f5a1659113f913fe152b02a58d95a5e7b38 100644
--- a/tangostationcontrol/tangostationcontrol/devices/unb2.py
+++ b/tangostationcontrol/tangostationcontrol/devices/unb2.py
@@ -12,7 +12,7 @@
 """
 
 # PyTango imports
-from tango.server import run
+from tango.server import run, command
 from tango.server import device_property, attribute
 from tango import AttrWriteType
 # Additional import