Replace stage task with build