Can you drop the `device` param and just hardcode this in the agent for now?