Changed misplaced/mismerged piece of code to fix agent staging.