small fix for surival tasks
more data sets