updated migration for moving to new agent setup