"We're just working our way through a long legal process."
Asked whether he was comfortable with the length of time it was taking for Rizalman to be sent to New Zealand, Mr Key said: "We would prefer it was quicker but in the end I take a lot of store out of the personal assurance that Prime Minister Najib gave me.
"There's no reason for the Malaysians to make it any more difficult than it should be. He'll have his own legal rights but I'm confident he will return to New Zealand."
Mr Key said he could not give a timeframe for Rizalman's return.
The former adviser at the Malaysian Embassy was arrested on May 10 after allegedly following Tania Billingsley to her Brooklyn home and attacking her.
The Ministry of Foreign Affairs and Trade (Mfat) asked Malaysia to waive immunity so he could face the charges, but behind-the-scenes communication led Malaysia to believe he could return home.
An independent inquiry is looking at how Rizalman was allowed to leave New Zealand. Mfat says the inquiry is ongoing.