I am using ag-grid for my project. As shown in the image, once I sort IMEI or another column, the sorted values stay, but if I sort validity (it is sorting based on a custom sort function), it goes back to the default value.

This only happens in the production environment. I have observed that there is some issue with ag-grid after react app is built. Any suggestions on how to make it work?

Here is the custom sorting function (for reference):

comparator: (valueA, valueB) => {
        if (valueA === valueB) return 0;
        return valueA > valueB ? 1 : -1;
      },

Click here to see image of the table