Small update to build task.