Fix columnar driver tests