Modify the way to import device module;