Sorting when using md-virtual-repeat and md-on-demand

Hello i'm trying to build an md-data-table that using an md-virtual repeat-container and get items using md-on-demand. However when doing that the column sorting stops working. Can you please suggest how to go about fixing this?

Thanks