Update automation-code

Fixing some copying issues.