[Q] Which headsets work with the Mi 5?

So I've had the Mi 5 for over a month now and I've been using a Samsung headset which worked fine (the volume controls don't seem to function correctly in Resurrection Remix). However I bought a Sony headset yesterday (MDREX450AP) and the earphones don't seem to work with the phone, however the headset button does work. Has anyone else come across anything similar, has anyone tried Apple earbuds with the Mi 5? Just for the record, I'm not looking to get Apple earbuds but I'm looking on Amazon and the headsets for Apple phones seem to be cheaper than the ones for Samsung/Android.

Thanks in advance,
Aaron :)