Use default_device

@zarate noticed the code is broken on #220