add missing import in device steps