get_nz_map_ctx_from_stats() is take TX_CLASS instead of TX_TYPE, this
commit makes sure right type of parameter is passed in.

Change-Id: I09b97a31890050189fc12f0133c0bde83fc79615