IO test dats updated

Fixed torch bug