Heatmap tickvals and ticktext not working

I am trying to use manual labels for my heatmap plot but it doesn’t seem to work (it works in surface plot though).

var layout = {
  title: 'Model Visualization',
  autosize: false,
  width: 500,
  height: 500,
  margin: {
    l: 0,
    r: 0,
    b: 50,
    t: 50,
    pad: 2
  },
  scene:{
      xaxis: {tickvals: row,
               ticktext : rowS,
      },
      yaxis: {tickvals: column,
               ticktext : columnS,
      },
}
};

what am I doing wrong ?

var layout = {
  title: 'Model Visualization',
  autosize: false,
  width: 500,
  height: 500,
  margin: {
    l: 0,
    r: 0,
    b: 50,
    t: 50,
    pad: 2
  },
      xaxis: {tickvals: row,
               ticktext : rowS,
      },
      yaxis: {tickvals: column,
               ticktext : columnS,
      },
};

should work. That scene container has no effect on non-3D graphs