Fixed assign issue in device_vector