This also needs a change to `get_default_device()`.