Set correct device