I have the same problem and I've tried many custom ROMs including MIUI, but the problem still persisted.
But I've found a simple solution to this problem, I've changed display off setting to 5 min.
setting>> display>> sleep>> After 5 minutes of inactivity.
This works for the calls less than 5 minutes.
Please click thanks if it works for you.