Fixed import error in device setup