Fixed unittest after the device changes