Should this be moved to `device` then?