Fixed missing data branch in BaselineTask